Mazz18 commited on
Commit
81381b7
·
verified ·
1 Parent(s): bb98dfa
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -1,11 +1,28 @@
1
  import gradio as gr
2
  import random
 
3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  def echo(message, history):
5
- choices = ["yes", "No", "As I see it", "yes", "Outlook good", "Yes - definitely", "without a doubt", "ask again later", "cannot predict now", "my reply is no", "very doubtful", ]
6
  return random.choice(my_list)
7
 
8
 
9
- chatbot = gr.ChatInterface(echo, type="messages", title = "Eight Ball", )
10
 
11
  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():
8
+
9
+ messages = [{"role":"system", "content": "You are a freindly 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(response["choices"][0]["message"]["content"].strip())
18
+
19
+ return response["choices"][0]["message"]["content"].strip()
20
+
21
  def echo(message, history):
22
+ my_list = ["yes", "No", "As I see it", "yes", "Outlook good", "Yes - definitely", "without a doubt", "ask again later", "cannot predict now", "my reply is no", "very doubtful", ]
23
  return random.choice(my_list)
24
 
25
 
26
+ chatbot = gr.ChatInterface(respond, type="messages")
27
 
28
  chatbot.launch()