update generator

generator.py

@@ -59,7 +59,7 @@ def run_model(input_string, **generator_args):
     
   inputs = tokenize([input_string])
     
-  res = model.generate(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'], **generate_wargs)
+  res = model.generate(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'], **generate_args)
   output = hftokenizer.decode(res[0], skip_special_tokens=True)
   output = [item.split("<sep>") for item in output]
   output = [o.strip() for o in output[:-1]]
@@ -79,7 +79,7 @@ def QA(question, context):
       'context':context
   }
   res = nlp(format)
-  output = f"{question}\n{string.capwords(res['answer'])}\tscore : [{res['score']}] \n"
+  output = f"{question}\n{string.capwords(res['answer'])}\n"
   return output
   # inputs = tokenizer(question, context, return_tensors="pt")
   # # Run the model, the deepset way