Update app.py

app.py

@@ -28,13 +28,13 @@ Here is the relevant evidence from the database:
 
 Answer this question based only on the evidence above:
 {question}"""
-    
-    response = client.text_generation(
-        prompt,
+
+    response = client.chat_completion(
+        messages=[{"role": "user", "content": prompt}],
         model="mistralai/Mistral-7B-Instruct-v0.3",
-        max_new_tokens=512
+        max_tokens=512
     )
-    return response
+    return response.choices[0].message.content
 
 demo = gr.Interface(
     fn=answer_question,