Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import random | |
| from huggingface_hub import InferenceClient | |
| # import lines go at the top: any libraries I need to import go up here ^^ | |
| 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) | |
| return response['choices'][0]['message']['content'].strip() | |
| def echo(message, history): | |
| return message | |
| def yes_or_no(message, history): | |
| return random.choice(['Yes', 'No', 'Maybe', 'Ask Again']) | |
| chatbot = gr.ChatInterface(respond, type = "messages") | |
| # defining my chatbot so that the user can interact and see their conversation history and send new messages | |
| chatbot.launch() |