Update app.py

app.py

@@ -1 +1,19 @@
-print("Hlw World")
+import streamlit as st
+import ollama
+
+def main():
+    st.title("Llama Chatbot")
+    
+    user_input = st.text_input("You:", "Why is the sky blue?")
+
+    if st.button("Ask"):
+        with st.spinner("Thinking..."):
+            response = ollama.chat(model='llama2', messages=[{
+                'role': 'user',
+                'content': user_input,
+            }])
+        st.success("Llama says:")
+        st.write(response['message']['content'])
+
+if __name__ == "__main__":
+    main()