Update app.py

app.py

@@ -1,12 +1,10 @@
 from langchain_community.llms import Ollama
 import gradio as gr
 
-def generate(name):
-	llm = Ollama(model="openchat")
+def generate(user_input):
+    llm = Ollama(model="openchat")
+    response = llm.invoke(user_input)
+    return response
 
-	response = llm.invoke(name)
-
-	return response
-
-iface = gr.ChatInterface(fn=generate)
-iface.launch()
+iface = gr.Interface(fn=generate, inputs="text", outputs="text", title="Ollama Chat", description="Chat with Ollama by entering your message.")
+iface.launch()