First-chatbot / app.py
antoinette-f's picture
changed random yes/no chatbot into general chatbot
ea51f49 verified
raw
history blame
683 Bytes
import gradio as gr
from huggingface_hub import InferenceClient
client = InferenceClient("microsoft/phi-4")
def respond(message,history): #function to return the history of messages sent
messages[{"role": "system", "content": "You are a friendly chatbot!"}]
if history:
messages.extend(history)
messages.append({"role": "user", "content":message})
responses = client.chat_completion(
messages,
max_token = 568
)
return response['choices'][0]['message']['content'].strip()
chatbot = gr.ChatInterface(respond, type="messages", title = "Friendly chatbot")
# chatbot UI - conversation history and user input
chatbot.launch()