| |
| |
|
|
| import gradio as gr |
| from huggingface_hub import InferenceClient |
|
|
| client = InferenceClient("Qwen/Qwen2.5-7B-Instruct-1M") |
|
|
| def respond(message, history): |
| messages = [{"role": "system", "content": "I am a kind chatbot."}] |
|
|
| |
| if history: |
| for user_msg, assistant_msg in history: |
| messages.append({"role": "user", "content": user_msg}) |
| messages.append({"role": "assistant", "content": assistant_msg}) |
|
|
| |
| messages.append({"role": "user", "content": message}) |
|
|
| |
| |
| |
| response = "" |
|
|
| |
| for message in client.chat_completion( |
| messages, |
| max_tokens=500, |
| temperature=.1, |
| stream=True): |
|
|
| |
| token = message.choices[0].delta.content |
| response += token |
| yield response |
|
|
| chatbot = gr.ChatInterface(respond) |
|
|
| chatbot.launch() |