practice for lesson 11 generative AI

app.py

@@ -1,12 +1,24 @@
 import gradio as gr
 import random
+from huggingface_hub import InferenceClient
 
-def random_message(message, history):
+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 echo(message, history):
     choices = ["yes", "no", "sure", "absolutely", "of course not", "by no means"]
     chat_answer = random.choice(choices)
     #use random to select ones of those choices
     return chat_answer
 
-chatbot = gr.ChatInterface(random_message, type = 'messages', title = "KWK Chatbot")
+chatbot = gr.ChatInterface(respond, type = 'messages')
 
 chatbot.launch()