shreyasa09 commited on
Commit
e64e7c1
·
verified ·
1 Parent(s): 2fbee9c

Practice for lesson 11 generative AI

Browse files
Files changed (1) hide show
  1. app.py +22 -2
app.py CHANGED
@@ -1,8 +1,28 @@
1
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  def echo(messaged, history):
4
- return messaged
 
 
 
5
 
6
- chatbot = gr.ChatInterface(echo)
7
 
8
  chatbot.launch()
 
1
  import gradio as gr
2
+ import random
3
+ from huggingface_hub import InferenceClient
4
+
5
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
6
+
7
+ def respond(message, history):
8
+
9
+ messages = [{"role": "system", "content": "You are a friendly chatbot."}]
10
+ if history:
11
+ messages.extend(history)
12
+
13
+ messages.append({"role : user", "content" : message})
14
+
15
+ response = client.chat_completion(messages, max_tokens = 100)
16
+
17
+ print (return response ["choices"] (0)["message"]["content"].strip())
18
+
19
+ return response ["choices"] (0)["message"]["content"].strip()
20
 
21
  def echo(messaged, history):
22
+ my_list = ["sure", "ask again", "maybe"]
23
+ return random.choice(my_list)
24
+ chatbot = gr.ChatInterface(respond, type = "messages")
25
+
26
 
 
27
 
28
  chatbot.launch()