chatbotlab / app.py
bhavya-k's picture
try
57504c3 verified
raw
history blame
1.01 kB
import gradio as gr
import random
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)
print(response ["choices"][0]["message"]["content"].strip())
return response ["choices"][0]["message"]["content"].strip()
def random_message(message, history):
choices = ["you have 24 hours", "yay ur lucky or whateva", "do not step outside", "try again", "pop quiz days be like", "please tell me you're not asking about a man.", "my sources tell me no", "haha you thought", "Im like 90.9% sure your being watched"]
random_element = random.choice(choices)
return random_element
chatbot = gr.ChatInterface(respond, type="messages")
chatbot.launch()