File size: 550 Bytes
588bdc0
bda7c26
588bdc0
fd92139
 
 
 
 
 
 
 
 
0e4187f
fd92139
588bdc0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")

def respond(message,history):
    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=0.2)
    return response['choices'][0]['message']['content'].strip()
    
chatbot = gr.ChatInterface(respond, type="messages")
chatbot.launch()