itzjunayed commited on
Commit
2e6f5fa
·
verified ·
1 Parent(s): de07c6c

added token

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,10 +1,11 @@
 
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("meta-llama/Llama-2-7b-chat-hf")
8
 
9
  def respond(
10
  message,
 
1
+ import os
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
 
5
+ # Read token from environment variable
6
+ HF_TOKEN = os.getenv("HF_TOKEN")
7
+
8
+ client = InferenceClient(model="meta-llama/Llama-2-7b-chat-hf", token=HF_TOKEN)
9
 
10
  def respond(
11
  message,