leolaish commited on
Commit
8b3c15a
·
verified ·
1 Parent(s): 439d527

Update app1.py

Browse files
Files changed (1) hide show
  1. app1.py +22 -0
app1.py CHANGED
@@ -1,3 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  demo = gr.ChatInterface(
2
  respond,
3
  title="MediPro",
 
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, system_message, max_tokens, temperature, top_p):
7
+ messages = [{"role": "system", "content": system_message}]
8
+ for val in history:
9
+ if val[0]:
10
+ messages.append({"role": "user", "content": val[0]})
11
+ if val[1]:
12
+ messages.append({"role": "assistant", "content": val[1]})
13
+ messages.append({"role": "user", "content": message})
14
+ response = ""
15
+ for message in client.chat_completion(messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p):
16
+ token = message.choices[0].delta.content
17
+ response += token
18
+ yield response
19
+
20
+ def clear_history():
21
+ return [], ""
22
+
23
  demo = gr.ChatInterface(
24
  respond,
25
  title="MediPro",