Create app.py

app.py

@@ -0,0 +1,29 @@
+import streamlit as st
+from transformers import pipeline
+
+st.set_page_config(page_title="SUST Math AI", page_icon="🎓")
+
+# This loads the model directly on Hugging Face's server
+@st.cache_resource
+def load_solver():
+    return pipeline("text-generation", model="HuggingFaceTB/SmolLM2-1.7B-Instruct")
+
+pipe = load_solver()
+
+st.title("🎓 SUST B.Sc. Math Solver")
+st.markdown("Enter your problem below to get a step-by-step solution.")
+
+user_query = st.chat_input("Ex: Find the derivative of (x^2 + 1)^3")
+
+if user_query:
+    with st.chat_message("user"):
+        st.write(user_query)
+
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            messages = [
+                {"role": "system", "content": "You are a professional Mathematics Professor. Solve the problem step-by-step with clear logic."},
+                {"role": "user", "content": user_query},
+            ]
+            result = pipe(messages, max_new_tokens=500, do_sample=False)
+            st.markdown(result[0]['generated_text'][-1]['content'])