SPACEBOT-DEMO / app.py
sdsiddharth's picture
Update app.py
4ee9078 verified
Raw
History Blame Contribute Delete
472 Bytes
import gradio as gr
import ollama
def chat_with_spacebot(message):
response = ollama.chat(model="sdsiddharth/SPACEBOT", messages=[{"role": "user", "content": message}])
return response['message']['content']
iface = gr.Interface(
fn=chat_with_spacebot,
inputs=gr.Textbox(label="Ask SPACEBOT"),
outputs=gr.Textbox(label="Response"),
title="🚀 Chat with SPACEBOT",
description="Try SPACEBOT instantly without installation!",
)
iface.launch()