Update app.py

app.py

@@ -66,26 +66,21 @@ import streamlit as st
 #from llama_index.llms.ollama import Ollama
 from myollama import Ollama
 
-# Initialize the Ollama model
-llm = Ollama(model="llama3", request_timeout=120.0)
+def main():
+    st.title("Ollama LLM Demo")
+    
+    # Get user input
+    prompt = st.text_input("Enter your prompt:", "What is the capital of France?")
+    
+    if st.button("Get Response"):
+        # Create an instance of the Ollama class
+        llm = Ollama(model="llama2", request_timeout=60.0)
+        
+        # Get the response from the LLM
+        response = llm.complete(prompt)
+        
+        # Display the response
+        st.write(response)
 
-st.title("Paul Graham Information Fetcher")
-
-# Text input for the query
-query = st.text_input("Enter your query:", value="Who is Paul Graham?")
-
-# Button to trigger the API call
-if st.button("Get Response"):
-    with st.spinner("Fetching response..."):
-        try:
-            # Fetch the response from the model
-            resp = llm.complete(query)
-            # Display the response
-            st.success("Response fetched successfully!")
-            st.write(resp)
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
-
-# Run the Streamlit app
 if __name__ == "__main__":
-    st.run()
+    main()