Spaces:

Subha95
/

Harry_potter_wiki

Sleeping

Subha95 commited on Sep 3, 2025

Commit

cff40c6

verified ·

1 Parent(s): 966597a

Update chatbot_rag.py

Files changed (1) hide show

chatbot_rag.py CHANGED Viewed

@@ -29,7 +29,7 @@ def build_qa():
     # 3. Load LLM (Phi-3 mini)
     print("🔹 Loading LLM...")
-    model_id = "tiiuae/falcon-7b-instruct"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id)

     # 3. Load LLM (Phi-3 mini)
     print("🔹 Loading LLM...")
+    model_id = "declare-lab/flan-alpaca-large"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id)