Update app.py

app.py

@@ -35,9 +35,9 @@ def fetch_answers(question, clincal_note ):
          if answer.startswith('.') or answer.startswith(':'):
              answer = answer[1:].strip()
          if answer in passage_sentences[i]:
-             evidence_sentence = passage_sentences[i]
+             evidence_sentence = evidence_sentence + " " + passage_sentences[i]
      
-         model_input = f"question: {query} context: {evidence_sentence}"
+         
          #encoded_input = tokenizer([model_input],
          #                            return_tensors='pt',
          #                            max_length=512,
@@ -46,11 +46,11 @@ 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)[0]['generated_text']
-         break
          
+     model_input = f"question: {query} context: {evidence_sentence}"
+     output_answer = text2text_generator(model_input)[0]['generated_text']
      result_str = "# ANSWER "+str(count)+": "+ output_answer +"\n"  
-     result_str = result_str + "REFERENCE: "+ answer + "\n\n"
+     result_str = result_str + "REFERENCE: "+ evidence_sentence + "\n\n"
      top_5_query_paragraph_answer_list += result_str
      count+=1