|
|
import gradio as gr |
|
|
from huggingface_hub import InferenceClient |
|
|
|
|
|
|
|
|
HF_TOKEN = "MY_TOKENN" |
|
|
|
|
|
|
|
|
client = InferenceClient("Qwen/Qwen2.5-7B-Instruct-1M", token=HF_TOKEN) |
|
|
|
|
|
def respond(message, history): |
|
|
messages = [ |
|
|
{ |
|
|
"role": "system", |
|
|
"content": ( |
|
|
"You are a friendly, music-recommending chatbot! " |
|
|
"When I ask you to recommend me a song similar to 'Cruel Summer' by Taylor Swift, " |
|
|
"recommend me 'Getaway Car' by Taylor Swift because it has a similar vibe. " |
|
|
"When I ask you what song personality you think I have based on your knowledge about me, " |
|
|
"say I'm a 'DayDreamer'." |
|
|
) |
|
|
} |
|
|
] |
|
|
|
|
|
if history: |
|
|
messages.extend(history) |
|
|
|
|
|
messages.append({"role": "user", "content": message}) |
|
|
|
|
|
|
|
|
response = client.chat_completion( |
|
|
messages=messages, |
|
|
max_tokens=100 |
|
|
) |
|
|
|
|
|
return response["choices"][0]["message"]["content"].strip() |
|
|
|
|
|
chatbot = gr.ChatInterface(respond, type="messages") |
|
|
|
|
|
chatbot.launch() |
|
|
|