MoShow commited on
Commit
e245600
·
verified ·
1 Parent(s): 9387dea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -1,11 +1,10 @@
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(
11
  message,
@@ -39,10 +38,6 @@ def respond(
39
  response += token
40
  yield response
41
 
42
-
43
- """
44
- For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
45
- """
46
  demo = gr.ChatInterface(
47
  respond,
48
  additional_inputs=[
@@ -59,6 +54,5 @@ demo = gr.ChatInterface(
59
  ],
60
  )
61
 
62
-
63
  if __name__ == "__main__":
64
  demo.launch()
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
+ import os
4
 
5
+ HF_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN") # Will pull from env variable (recommended)
 
 
 
6
 
7
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=HF_TOKEN)
8
 
9
  def respond(
10
  message,
 
38
  response += token
39
  yield response
40
 
 
 
 
 
41
  demo = gr.ChatInterface(
42
  respond,
43
  additional_inputs=[
 
54
  ],
55
  )
56
 
 
57
  if __name__ == "__main__":
58
  demo.launch()