Spaces:

shripadbhat
/

Document_Question_Answering

Runtime error

shripadbhat commited on Nov 7, 2022

Commit

0d9fb6f

1 Parent(s): e5f4b72

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def fetch_answers(question, clincal_note ):
          #output = model.generate(input_ids = encoded_input.input_ids,
          #                           attention_mask = encoded_input.attention_mask)
          #output_answer = tokenizer.decode(output[0], skip_special_tokens=True)
-         output_answer = text2text_generator(model_input)['generated_text']
          break
      result_str = "# ANSWER "+str(count)+": "+ output_answer +"\n"

          #output = model.generate(input_ids = encoded_input.input_ids,
          #                           attention_mask = encoded_input.attention_mask)
          #output_answer = tokenizer.decode(output[0], skip_special_tokens=True)
+         output_answer = text2text_generator(model_input)[0]['generated_text']
          break
      result_str = "# ANSWER "+str(count)+": "+ output_answer +"\n"