Spaces:

Subha95
/

Harry_potter_wiki

Sleeping

Subha95 commited on Sep 4, 2025

Commit

b6660a5

verified ·

1 Parent(s): 0ee7c79

Update chatbot_rag.py

Files changed (1) hide show

chatbot_rag.py CHANGED Viewed

@@ -40,7 +40,7 @@ def build_qa():
     # 3. Load LLM (Phi-3 mini)
     print("🔹 Loading LLM...")
-    model_id = "meta-llama/Llama-3.2-3B-Instruct"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,

     # 3. Load LLM (Phi-3 mini)
     print("🔹 Loading LLM...")
+    model_id = "meta-llama/Llama-3.2-1B"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model = AutoModelForCausalLM.from_pretrained(
         model_id,