test / app.py
flascala's picture
Update app.py
398cb62 verified
Raw
History Blame Contribute Delete
1.71 kB
import gradio as gr
from huggingface_hub import InferenceClient
client = InferenceClient("Qwen/Qwen2.5-7B-Instruct")
def respond(message, history):
messages = [{"role": "system", "content": "You are a gym bro that is passionate about meal prepping. Help people create fitness goals, create healthy life styles, and eat well. Responses should be under 600 words. Keep language simple and 'bro-y'.understand the user's goal before giving recommendations."}]
#content is the system message
if history:
messages.extend(history)
messages.append({"role": "user", "content": message})
response = "" #empty string and indent client.chat
stream = client.chat_completion( #turn into a stream to arrive piece by piece
messages,
max_tokens=600,
temperature=0.3, # Higher temperature values like 0.8 will make the output more random,
# while lower values like 0.2 will make it more focused and deterministic.
stream=True #pasted into client.chat
)
for message in stream: #add a for loop
token = message.choices[0].delta.content #paste in
response += token
yield response #change to yield and get rid of extra parts that are now in for loop
# Customize colors, sizing and fonts
custom_theme = gr.themes.Soft(
primary_hue="pink",
secondary_hue="fuchsia",
neutral_hue="gray",
spacing_size="lg",
radius_size="lg",
text_size="lg",
font=[gr.themes.GoogleFont("IBM Plex Sans"), "sans-serif"],
font_mono=[gr.themes.GoogleFont("IBM Plex Mono"), "monospace"]
)
chatbot = gr.ChatInterface(respond, type="messages", theme=custom_theme)
chatbot.launch()