File size: 602 Bytes
e1fdffd
fb0b429
 
e20c561
e1fdffd
e4db748
c8abf10
fb0b429
 
 
 
 
 
 
 
 
0b8d27f
fb0b429
 
 
e1fdffd
fb0b429
 
4a9302d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")

def respond (message, history):
    messages = [{"role": "system", "content": "You're a friendly chatbot"}]

    if history:
        messages.extend(history)

    messages.append({"role": "user", "content": message})

    response = client.chat_completion(
        messages, 
        max_tokens = 100, 
        temperature = 1.2 
    )
    
    return response['choices'][0]['message']['content'].strip()
    
chatbot = gr.ChatInterface(respond, type = 'messages')

chatbot.launch()