Update app.py

app.py

@@ -8,9 +8,6 @@ import rag
 # -----------------------------
 fine_tuning.load_and_train()
 
-# -----------------------------
-# Combined Answer Function
-# -----------------------------
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
@@ -19,7 +16,7 @@ def combined_generate(prompt, max_tokens):
         start_times = {"Fine-tuned": time.time(), "RAG": time.time()}
 
         futures = {
-            executor.submit(fine_tuning.generate_answer, tokenizer, device, prompt, max_tokens): "Fine-tuned",
+            executor.submit(fine_tuning.generate_answer, prompt, max_tokens): "Fine-tuned",
             executor.submit(rag.generate_answer, prompt): "RAG",
         }