bushra1dajam commited on
Commit
f92479c
·
verified ·
1 Parent(s): 6bb2626

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -33,8 +33,7 @@ with open('svm_model.pkl', 'rb') as f:
33
 
34
 
35
  model = AutoModelForSeq2SeqLM.from_pretrained("AraT5/model")
36
- tokenizer = BartTokenizer.from_pretrained('AraT5/tokenizer')
37
-
38
 
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
40
  model.to(device)
 
33
 
34
 
35
  model = AutoModelForSeq2SeqLM.from_pretrained("AraT5/model")
36
+ tokenizer = AutoTokenizer.from_pretrained('AraT5/tokenizer',use_fast=False)
 
37
 
38
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
39
  model.to(device)