File size: 763 Bytes
8df82ee
c383862
8df82ee
c383862
 
5ffbaca
c383862
 
 
 
 
 
 
 
 
 
 
 
 
 
f4777e7
c383862
 
 
 
8df82ee
3b75d7d
8df82ee
1d55cee
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
29
import gradio as gr
from huggingface_hub import InferenceClient

#create instance of InferenceClient
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")

#respond function
def respond (message, history):
    messages = [{"role": "system", "content": "You are a friendly color theory chatbot."}]

#include conversation history
    if history:
        messages.extend(history)

#add current user's message
    messages.append({"role":"user", "content": message})

#chat completion call
    response = client.chat_completion(
        messages,
        max_tokens=512
    )

#extract and return chatbots response
    return response['choices'][0]['message']['content'].strip()
    
chatbot = gr.ChatInterface(respond, type="messages")

chatbot.launch(debug=True)