Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 from concurrent.futures import ThreadPoolExecutor
 import fine_tuning
-from fine_tuning import load_and_train
+import rag
 
 # -----------------------------
 # Load fine-tuned model
 # -----------------------------
-model, tokenizer, device = load_and_train()
+model, tokenizer, device = fine_tuning.load_and_train()
 
 # -----------------------------
 # RAG Backend (Stub Example)
@@ -31,7 +31,7 @@ def combined_generate(prompt, max_tokens):
 
         futures = {
             executor.submit(fine_tuning.generate_answer, model, tokenizer, device, prompt, max_tokens): "Fine-tuned",
-            executor.submit(generate_answer_rag, prompt, max_tokens): "RAG",
+            executor.submit(rag.generate_answer, prompt): "RAG",
         }
 
         answers = {"Fine-tuned": "", "RAG": ""}