abdullahmeda commited on
Commit
31466c9
·
verified ·
1 Parent(s): 6631744

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id)
17
  print("Loading NLTL & and scikit-learn model...")
18
  NLTK = nltk_load('data/english.pickle')
19
  sent_cut_en = NLTK.tokenize
20
- clf = joblib.load(f'data/gpt2-large-model', 'rb')
21
 
22
  CROSS_ENTROPY = torch.nn.CrossEntropyLoss(reduction='none')
23
 
 
17
  print("Loading NLTL & and scikit-learn model...")
18
  NLTK = nltk_load('data/english.pickle')
19
  sent_cut_en = NLTK.tokenize
20
+ clf = joblib.load(f'data/gpt2-large-model')
21
 
22
  CROSS_ENTROPY = torch.nn.CrossEntropyLoss(reduction='none')
23