Update app.py

app.py

@@ -40,10 +40,15 @@ def generate_response(model, tokenizer, prompt, max_length=512, num_return_seque
 
 # Define a função para responder às perguntas
 def answer_question(context, question):
-    if context:
-        prompt = f"Com base no contexto fornecido, responda a questão de forma sucinta.\nContexto:\n{context}\nQuestão:\n{question}\n### Resposta:"
-    else:
-        prompt = f"Questão:\n{question}\n### Resposta:"
+    prompt = f"""Com base no contexto fornecido, responda a questão de forma sucinta.
+    Contexto:
+    {context}
+    Questão:
+    {question}
+    ### Resposta:
+    """
+    #else:
+        #prompt = f"Questão:\n{question}\n### Resposta:"
     return generate_response(model, tokenizer, prompt)