File size: 707 Bytes
88563c7
 
 
95a9cac
 
88563c7
587d127
88563c7
 
95a9cac
88563c7
 
 
3e2fd09
88563c7
 
 
48043d0
88563c7
eb4ac7a
942f5ad
587d127
 
88563c7
 
 
95a9cac
 
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
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=2, top_p=1)
    #temperature and top_p control randomness
    
    return response['choices'][0]['message']['content'].strip()
        
chatbot = gr.ChatInterface(respond, type="messages")
chatbot.launch()