Spaces:

ammoncoder123
/

IPTchatbot

Running

ammoncoder123 commited on Jan 12

Commit

ca6f81c

verified ·

1 Parent(s): 2d72ee6

Update chatbot.py

Files changed (1) hide show

chatbot.py CHANGED Viewed

@@ -9,7 +9,7 @@ login(token=os.getenv("HF_TOKEN"))
 # ================= CACHE THE MODEL =================
 @st.cache_resource
 def load_model():
-    model_id = "ammoncoder123/IPTchatbotModel1-1.7B"
     st.write("Loading tokenizer...")
     tokenizer = AutoTokenizer.from_pretrained(model_id)

 # ================= CACHE THE MODEL =================
 @st.cache_resource
 def load_model():
+    model_id = "ammoncoder123/IPTchatbotModel-1.7B"
     st.write("Loading tokenizer...")
     tokenizer = AutoTokenizer.from_pretrained(model_id)