Update app.py

app.py

@@ -30,18 +30,13 @@ client = InferenceClient("google/mt5-small", token=os.getenv("HUGGINGFACEHUB_API
 ranker = Reranker("answerdotai/answerai-colbert-small-v1", model_type='colbert')
 
 def generate_text(context, query):
-    prompt = f"""
-        Contesta la siguiente pregunta basándote solo en el contexto proporcionado. 
-        Si no puedes encontrar la respuesta en el contexto, indica que no tienes suficiente información.
-        
-        Contexto:
-        {context}
-        
-        Pregunta:
-        {query}
-        
-        Respuesta:
-        """
+    prompt = f"""Responde en español. Basándote en el siguiente contexto contesta de forma clara y directa.
+
+    Contexto:
+    {context}
+    
+    Pregunta: {query}
+    Respuesta:"""
     response = client.text_generation(prompt=prompt, max_new_tokens=200)
     return response.strip()