Update chatbot_rag.py

chatbot_rag.py

@@ -29,9 +29,9 @@ def build_qa():
 
     # 3. Load LLM (Phi-3 mini)
     print("🔹 Loading LLM...")
-    model_id = "microsoft/Phi-3-mini-4k-instruct"
+    model_id = "google/flan-t5-small"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
-    model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_id, device_map="auto")
 
     pipe = pipeline(
         "text-generation",