File size: 765 Bytes
fae3148 005d889 1e605f9 d324bcf 8351e16 d324bcf 1e605f9 a476719 d324bcf 237bab6 d324bcf 5fa2ece 8351e16 d324bcf a476719 d324bcf 0a87c7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from huggingface_hub import InferenceClient
import gradio as gr
import random
client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
def respond(message, history):
# responses = ["Yes", "No"]
# return random.choice(responses)
messages = [
{"role":"system",
"content": "you are a friendly chatbot"}
]
if history:
messages.extend(history)
messages.append(
{"role": "user",
"content": message}
)
response = client.chat_completion(
messages, max_tokens = 100, temperature=1.3, top_p=.2
#temp and top_p control randomness
)
return response['choices'][0]['message']['content'].strip()
chatbot = gr.ChatInterface(respond, type="messages")
chatbot.launch() |