Subha95 commited on
Commit
ec8075b
·
verified ·
1 Parent(s): de1a78c

Update chatbot_rag.py

Browse files
Files changed (1) hide show
  1. chatbot_rag.py +1 -1
chatbot_rag.py CHANGED
@@ -40,7 +40,7 @@ def build_qa():
40
  # 3. Load LLM (Phi-3 mini)
41
  print("🔹 Loading LLM...")
42
 
43
- model_id = "google/gemma-1.1-2b-it"
44
 
45
  tokenizer = AutoTokenizer.from_pretrained(model_id)
46
 
 
40
  # 3. Load LLM (Phi-3 mini)
41
  print("🔹 Loading LLM...")
42
 
43
+ model_id = "microsoft/phi-2"
44
 
45
  tokenizer = AutoTokenizer.from_pretrained(model_id)
46