from langchain_community.llms import Ollama
import gradio as gr

def generate(user_input):
    llm = Ollama(model="openchat")
    response = llm.invoke(user_input)
    return response

iface = gr.Interface(fn=generate, inputs="text", outputs="text", title="Ollama Chat", description="Chat with Ollama by entering your message.")
iface.launch()