Update app.py

app.py

@@ -265,9 +265,12 @@ def qa_system(method, question):
                                                temperature=0.7, do_sample=True,
                                                pad_token_id=tokenizer.eos_token_id)
         answer = tokenizer.decode(outputs[0], skip_special_tokens=True).split("Answer:")[-1].strip()
+        if "Question:" in answer:
+            answer = text.split("Question:")[0].strip()
+        
         end_time = time.time()
         response_time = round(end_time - start_time, 2)
-        return f"**Method:** Fine-Tuned Model", 0.95, response_time, answer
+        return f"**Method:** Fine-Tuned Model", 0.95, response_time, answer.strip()
 
 # -------------------------------
 # Gradio UI (same as before)