Spaces:
Sleeping
Sleeping
File size: 829 Bytes
3e4b803 938da7e a3dd323 3e4b803 cecc48f 3e4b803 a3dd323 9a6d6a0 a3dd323 938da7e a3dd323 3e4b803 | 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 30 31 32 33 34 35 | import gradio as gr
import random
import os
from huggingface_hub import InferenceClient
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
#def respond_yes_no_random(message, history):
# responses = ["yes", "no"]
# return random.choice(responses)
#chatbot = gr.ChatInterface(respond_yes_no_random, type = "messages", examples = ["HELLO", "Bye!"], title = "Annie's ChatBot")
def respond(message, history):
messages = [{"role": "system", "content": "You are a vampire!"}]
if history:
messages.extend(history)
messages.append({"role":"user","content": message})
response = client.chat_completion(
messages,
max_tokens=100
)
return response['choice'][0]['message']['content'].strip()
chatbot = gr.ChatInterface(respond, type = "messages")
chatbot.launch()
|