vinayakdev commited on
Commit
8336e59
1 Parent(s): bcad6bc

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +2 -2
generator.py CHANGED
@@ -39,9 +39,9 @@ def load_model():
39
  model = pickle.load(open('model.sav','rb'))
40
  tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  # return hfm, hft,tok, model
42
- return hfm, hft, tokenzer, model
43
 
44
- hfmodel, hftokenizer, tok, model = load_model()
45
 
46
  def run_model(input_string, **generator_args):
47
  generator_args = {
 
39
  model = pickle.load(open('model.sav','rb'))
40
  tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  # return hfm, hft,tok, model
42
+ return hfm, hft, tok, model
43
 
44
+ hfmodel, hftokenizer, tokenizer, model = load_model()
45
 
46
  def run_model(input_string, **generator_args):
47
  generator_args = {