Hugging Face's logo

Spaces:
vinayakdev
/
qa-generator
Runtime error

App Files Community
vinayakdev commited on
Commit
7ba4ac5
·
1 Parent(s): 07151db

Change electra to att

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -37,7 +37,7 @@ def load_model():
37
  hfm = pickle.load(open('hfmodel.sav','rb'))
38
  hft = T5TokenizerFast.from_pretrained("t5-base")
39
  model = pickle.load(open('model.sav','rb'))
40
- tok = et.from_pretrained("ahotrod/albert_xxlargev1_squad2_512 ")
41
  # return hfm, hft,tok, model
42
  return hfm, hft,tok, model
43
 
 
37
  hfm = pickle.load(open('hfmodel.sav','rb'))
38
  hft = T5TokenizerFast.from_pretrained("t5-base")
39
  model = pickle.load(open('model.sav','rb'))
40
+ tok = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512 ")
41
  # return hfm, hft,tok, model
42
  return hfm, hft,tok, model
43