Spaces:

ammoncoder123
/

IPTchatbot

Running

ammoncoder123 commited on Jan 12

Commit

c0db93e

verified ·

1 Parent(s): 156fb9c

Update chatbot.py

Files changed (1) hide show

chatbot.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline
 @st.cache_resource
 def load_model():
     # Use a public 7B model (excellent quality)
-    return pipeline("text-generation", model="your-ammoncoder123/IPTchatbot")
 pipe = load_model()

 @st.cache_resource
 def load_model():
     # Use a public 7B model (excellent quality)
+    return pipeline("text-generation",model_id = "ammoncoder123/IPTchatbotModel1-1.7B")
 pipe = load_model()