Ambereen commited on
Commit
948f6d6
·
verified ·
1 Parent(s): b13c626

adding huggingface & transitioning to the main branch

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -1,8 +1,21 @@
1
- import random
2
  import gradio as gr
 
 
 
 
 
 
 
 
 
3
 
4
- def random_response(message, history):
5
- return random.choice(["Yes", "No",'maybe','TRY AGAIN'])
 
 
 
 
6
 
7
  demo = gr.ChatInterface(random_response, type="messages", autofocus=False, theme='mgetz/Celeb_glitzy')
8
 
 
1
+
2
  import gradio as gr
3
+ from huggingface_hub import InferenceClient
4
+ client=InferenceClient("HuggingFaceH4/zephyr-7b-beta")
5
+ def respond(message, history):
6
+ messages= [{'role':'system', "content":"You are a friend chatbot"}]
7
+
8
+ if history:
9
+ messages.extend(history)
10
+
11
+ messages.append({'role':'user','content':meesage})
12
 
13
+ response = client.chat_completion(
14
+ messages,
15
+ max_tokens=100
16
+ temperature=0.2
17
+ )
18
+ return response['choices'][0]['message']['content'].strip()
19
 
20
  demo = gr.ChatInterface(random_response, type="messages", autofocus=False, theme='mgetz/Celeb_glitzy')
21