prempranavreddy commited on
Commit
786376c
·
verified ·
1 Parent(s): 9abef1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -1,10 +1,16 @@
1
  import gradio as gr
 
2
  from huggingface_hub import InferenceClient
3
 
4
- """
5
- For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
6
- """
7
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 
 
 
8
 
9
 
10
  def respond(
 
1
  import gradio as gr
2
+ import os
3
  from huggingface_hub import InferenceClient
4
 
5
+ HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
6
+
7
+ client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta", token=HF_TOKEN)
8
+
9
+
10
+ def respond(message):
11
+ for response in client.chat_completion(messages=[{"role": "user", "content": message}]):
12
+ yield response["choices"][0]["message"]["content"]
13
+
14
 
15
 
16
  def respond(