Lhumpal commited on
Commit
4a1495c
·
verified ·
1 Parent(s): 7f349bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -27,7 +27,15 @@ from huggingface_hub import InferenceClient
27
 
28
  app = FastAPI()
29
 
30
- client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
 
 
 
 
 
 
31
 
32
  class ChatRequest(BaseModel):
33
  message: str
 
27
 
28
  app = FastAPI()
29
 
30
+ # Get the token from the environment variable
31
+ hf_token = os.environ.get("HF_TOKEN")
32
+
33
+ if hf_token:
34
+ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
35
+ else:
36
+ raise ValueError("HF_TOKEN environment variable not set. Please add it as a secret in your Hugging Face Space.")
37
+
38
+ # Rest of your code...
39
 
40
  class ChatRequest(BaseModel):
41
  message: str