Hugging Face's logo

Spaces:
vinayakdev
/
qa-generator
Runtime error

App Files Community
vinayakdev commited on
Commit
6d17167
•
1 Parent(s): d414d6e

Happy Hugging Face!

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -56,7 +56,7 @@ def run_model(input_string, **generator_args):
56
  # al_tokenizer = att.from_pretrained("deepset/electra-base-squad2")
57
  # al_model = amqa.from_pretrained("deepset/electra-base-squad2")
58
 
59
- tokenizer = AutoTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
60
 
61
  model = AutoModelForQuestionAnswering.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
62
  # al_model = pickle.load(open('models/al_model.sav', 'rb'))
 
56
  # al_tokenizer = att.from_pretrained("deepset/electra-base-squad2")
57
  # al_model = amqa.from_pretrained("deepset/electra-base-squad2")
58
 
59
+ tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
60
 
61
  model = AutoModelForQuestionAnswering.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
62
  # al_model = pickle.load(open('models/al_model.sav', 'rb'))