Spaces:

ammoncoder123
/

IPTchatbot

Running

ammoncoder123 commited on Jan 12

Commit

1d8da75

verified ·

1 Parent(s): b64b83d

Update chatbot.py

Files changed (1) hide show

chatbot.py CHANGED Viewed

@@ -10,7 +10,7 @@ login(token=os.getenv("HF_TOKEN"))
 # ================= CACHE THE MODEL =================
 @st.cache_resource
 def load_model():
-    model_id = "ammoncoder123/IPTchatbotModel1-1.7B"  # Your correct repo
     quantization_config = BitsAndBytesConfig(
         load_in_4bit=True,

 # ================= CACHE THE MODEL =================
 @st.cache_resource
 def load_model():
+    model_id = "ammoncoder123/IPTchatbotModel-1.7B"  # Your correct repo
     quantization_config = BitsAndBytesConfig(
         load_in_4bit=True,