SarmaHighOnAI commited on
Commit
5f063dc
·
verified ·
1 Parent(s): 3ebd60c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from huggingface_hub import InferenceClient
4
 
5
  app = FastAPI()
6
- client = InferenceClient(token="YOUR_HF_TOKEN_HERE")
7
 
8
  class Request(BaseModel):
9
  prompt: str
 
1
+ import os
2
  from fastapi import FastAPI
3
  from pydantic import BaseModel
4
  from huggingface_hub import InferenceClient
5
 
6
  app = FastAPI()
7
+ client = InferenceClient(token=os.environ.get("HF_TOKEN"))
8
 
9
  class Request(BaseModel):
10
  prompt: str