Spaces:
Sleeping
Sleeping
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) |