vinayakdev commited on
Commit
24f2c54
1 Parent(s): 88f594a

Check again

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -57,7 +57,7 @@ def run_model(input_string, **generator_args):
57
 
58
  input_string = "generate questions: " + input_string + " </s>"
59
 
60
- inputs = tokenize(input_string)
61
 
62
  res = model.generate(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'], **generate_wargs)
63
  output = hftokenizer.decode(res[0], skip_special_tokens=True)
 
57
 
58
  input_string = "generate questions: " + input_string + " </s>"
59
 
60
+ inputs = tokenize([input_string])
61
 
62
  res = model.generate(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'], **generate_wargs)
63
  output = hftokenizer.decode(res[0], skip_special_tokens=True)