Update app.py

app.py

@@ -1,7 +1,21 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+chat = pipeline("text-generation", model="LeoLM/leo-mistral-hessianai-7b-chat")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+def chat_with_model(prompt):
+    response = chat(prompt, max_length=50)
+    return response[0]['generated_text']
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=chat_with_model,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Type your message here..."),
+    outputs="text",
+    title="Chat with LLM",
+    description="Type a message and chat with a language model."
+)
+
+if __name__ == "__main__":
+    iface.launch()