Hugging Face's logo

Spaces:
vinayakdev
/
qa-generator
Runtime error

App Files Community
vinayakdev commited on
Commit
d414d6e
•
1 Parent(s): 54db8a6

Happy Hugging Face!

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -34,7 +34,7 @@ import pickle
34
  # hfmodel = pickle.load(open('models/hfmodel.sav', 'rb'))
35
  hfmodel = alwm.from_pretrained("ThomasSimonini/t5-end2end-question-generation")
36
 
37
- hftokenizer = T5TokenizerFast.from_pretrained("ThomasSimonini/t5-end2end-question-generation")
38
  def run_model(input_string, **generator_args):
39
  generator_args = {
40
  "max_length": 256,
 
34
  # hfmodel = pickle.load(open('models/hfmodel.sav', 'rb'))
35
  hfmodel = alwm.from_pretrained("ThomasSimonini/t5-end2end-question-generation")
36
 
37
+ hftokenizer = T5TokenizerFast.from_pretrained("t5-base")
38
  def run_model(input_string, **generator_args):
39
  generator_args = {
40
  "max_length": 256,