Hugging Face's logo

Spaces:
vinayakdev
/
qa-generator
Runtime error

App Files Community
vinayakdev commited on
Commit
d0d8e41
·
1 Parent(s): 845c3bb

Rename hfm

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -40,7 +40,7 @@ def load_model():
40
  tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
42
  pickle.dump(model, open('model.sav', 'wb'))
43
- pickle.dump(hfmodel, open('hfmodel.sav', 'wb'))
44
 
45
  return hfm, hft,tok, model
46
  hfmodel, hftokenizer, tokenizer, model = load_model()
 
40
  tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
42
  pickle.dump(model, open('model.sav', 'wb'))
43
+ pickle.dump(hfm, open('hfmodel.sav', 'wb'))
44
 
45
  return hfm, hft,tok, model
46
  hfmodel, hftokenizer, tokenizer, model = load_model()