uzzam24 commited on
Commit
7658c25
·
1 Parent(s): 3176c9b

Force rebuild to install sentencepiece

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from transformers import XLNetTokenizer, XLNetModel
5
  import numpy as np
6
  import joblib
7
 
 
8
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
  tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')
10
  xlnet_model = XLNetModel.from_pretrained('xlnet-base-cased').to(device)
 
5
  import numpy as np
6
  import joblib
7
 
8
+
9
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
10
  tokenizer = XLNetTokenizer.from_pretrained('xlnet-base-cased')
11
  xlnet_model = XLNetModel.from_pretrained('xlnet-base-cased').to(device)