Update app.py

app.py

@@ -445,6 +445,10 @@ def generate_answer(prompt: str, tokenizer_f, model_f, max_length: int = 860, te
         print(f"Error generating answer: {e}")
         return "I apologize, but I encountered an error while generating the answer."
 
+def enhance_passage_with_entities(passage, entities):
+    return f"{passage}\n\nEntities: {', '.join(entities)}"
+
+
 def create_prompt(question: str, passage: str) -> str:
     return f"""As a medical expert, answer the following question based only on the provided passage. Be concise and direct.
 Passage: {passage}