vinayakdev commited on
Commit
2a22742
1 Parent(s): fcee375
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -68,7 +68,7 @@ def run_model(input_string, **generator_args):
68
  # al_tokenizer = pickle.load(open('models/al_tokenizer.sav', 'rb'))
69
  def QA(question, context):
70
  # model_name="deepset/electra-base-squad2"
71
- nlp = pipeline("question-answering",model=model,tok = "mrm8488/electra-small-finetuned-squadv2")
72
  format = {
73
  'question':question,
74
  'context':context
 
68
  # al_tokenizer = pickle.load(open('models/al_tokenizer.sav', 'rb'))
69
  def QA(question, context):
70
  # model_name="deepset/electra-base-squad2"
71
+ nlp = pipeline("question-answering",model=model,tokenizer = tok)
72
  format = {
73
  'question':question,
74
  'context':context