Spaces:

vinayakdev
/

qa-generator

Runtime error

vinayakdev commited on Jan 2, 2023

Commit

bcad6bc

•

1 Parent(s): 90350be

Update generator.py

Files changed (1) hide show

generator.py CHANGED Viewed

@@ -39,7 +39,7 @@ def load_model():
   model = pickle.load(open('model.sav','rb'))
   tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
   # return hfm, hft,tok, model
-  return hfm, hft,tok, model
 hfmodel, hftokenizer, tok, model = load_model()

   model = pickle.load(open('model.sav','rb'))
   tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
   # return hfm, hft,tok, model
+  return hfm, hft, tokenzer, model
 hfmodel, hftokenizer, tok, model = load_model()