irmakakin commited on
Commit
302f7cc
·
verified ·
1 Parent(s): 60fa7c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -17
app.py CHANGED
@@ -1,22 +1,24 @@
1
  import gradio as gr
2
- import random
3
  from huggingface_hub import InferenceClient
4
- client = InferenceClient('HuggingFaceH4/zephyr-7b-beta')
5
- #libraries^
6
- def respond(message,history):
 
 
7
  messages = [{"role": "system", "content": "You are a friendly chatbot."}]
8
- if history:
9
- messages.extend(history)
 
 
10
  messages.append({"role": "user", "content": message})
11
- response = client.chat_completion(
12
- messages,
13
- max_tokens=100
14
- )
15
- return response['choices'][0]['message']['content'].strip()
16
- #def echo(message, history):
17
- #return message
18
- #def yes_or_no(message, history):
19
- #return random.choice(['yes','no', 'maybe','try again', 'i dont know','hahahahahaha','guilty'])
20
 
21
- chatbot = gr.ChatInterface(respond, type = 'messages',title = '8 BALL', description = 'type a yes or no question',theme = 'soft')
22
- chatbot.launch()
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
 
2
  from huggingface_hub import InferenceClient
3
+
4
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
5
+
6
+ def respond(message, history):
7
+
8
  messages = [{"role": "system", "content": "You are a friendly chatbot."}]
9
+
10
+ if history:
11
+ messages.extend(history)
12
+
13
  messages.append({"role": "user", "content": message})
 
 
 
 
 
 
 
 
 
14
 
15
+ response = client.chat_completion(
16
+ messages,
17
+ max_tokens=100
18
+ )
19
+
20
+ return response['choices'][0]['message']['content'].strip()
21
+
22
+ chatbot = gr.ChatInterface(respond, type="messages")
23
+
24
+ chatbot.launch()