Lhumpal commited on
Commit
0d99dad
·
verified ·
1 Parent(s): 9c3b7f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ app = FastAPI()
9
 
10
  # Get the token from the environment variable
11
  hf_token = os.environ.get("HF_TOKEN")
 
12
 
13
  # client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
14
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
 
9
 
10
  # Get the token from the environment variable
11
  hf_token = os.environ.get("HF_TOKEN")
12
+ login(token=hf_token)
13
 
14
  # client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
15
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)