Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,15 @@ from huggingface_hub import InferenceClient
|
|
27 |
|
28 |
app = FastAPI()
|
29 |
|
30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|