Update app.py

app.py

@@ -252,6 +252,7 @@ def rag_pipeline(query: str):
 # Unified QA System
 # -------------------------------
 def qa_system(method, question):
+    start_time = time.time()
     if method == "Retrieval-Augmented Generation (RAG)":
         answer, confidence, elapsed = rag_pipeline(question)
         return f"**Method:** RAG-based Model", confidence, f"{elapsed} seconds", answer
@@ -264,7 +265,9 @@ 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()
-        return f"**Method:** Fine-Tuned Model", 0.95, "N/A", answer
+        end_time = time.time()
+        response_time = round(end_time - start_time, 2)
+        return f"**Method:** Fine-Tuned Model", 0.95, response_time, answer
 
 # -------------------------------
 # Gradio UI (same as before)