Update app.py

app.py

@@ -32,7 +32,6 @@ def fetch_answers(question, clincal_note ):
              answer = answer[1:].strip()
          if answer in passage_sentences[i]:
              evidence_sentence = passage_sentences[i]
-             break
      
          model_input = f"question: {query} context: {evidence_sentence}"
          encoded_input = tokenizer([model_input],
@@ -43,7 +42,8 @@ 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)
-
+         break
+         
      result_str = "# RESULT NO: "+str(count)+ output_answer +"\n"  
      result_str = result_str + "REFERENCE: "+ evidence_sentence + "\n\n"
      top_5_query_paragraph_answer_list += result_str