Connected chatbot to LLM

app.py

@@ -1,5 +1,21 @@
 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)
+
+    return response['choices'][0]['message']['content'].strip()
 
 def echo(message, history):
     return message
@@ -8,6 +24,6 @@ def yes_no(message, history):
     responses = ["Yes", "No"]
     return random.choice(responses)
     
-chatbot = gr.ChatInterface(yes_no)
+chatbot = gr.ChatInterface(respond, type="messages")
 
 chatbot.launch()