Practice for lesson 11 generative AI

app.py

@@ -1,11 +1,29 @@
 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 = ["Mhm", "nah fam", "I don't like your tone, buddy rewrite that.", "Uhm, sure, ig.", "What a good question! So-", "I know what you are.", "I'm uncomfortable with the energy we've created today.", "I like that question! Ask someone else."]
     random_choice = random.choice(choices)
     return random_choice
     
-chatbot = gr.ChatInterface(random_message, type = "messages", description = "<strong><center>Don't talk to me please.</strong><br>go away.", title = "I'm not a chatbot")
+chatbot = gr.ChatInterface(respond, type = "messages", description = "<strong><center>Don't talk to me please.</strong><br>go away.", title = "I'm not a chatbot")
 
 chatbot.launch()