Create chatbot.py

chatbot.py

@@ -0,0 +1,32 @@
+import streamlit as st
+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.title("My 1.7B Smart Chatbot")
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+for msg in st.session_state.messages:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
+
+if prompt := st.chat_input("Ask about IPT/ICT..."):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            response = pipe(prompt, max_new_tokens=300)[0]['generated_text']
+            st.markdown(response)
+
+    st.session_state.messages.append({"role": "assistant", "content": response})
+
+st.info("This is a 1.7B model demo — answers are generally accurate but verify important facts.")