waceke commited on
Commit
b712ab3
·
verified ·
1 Parent(s): 50f3ecd

Updated the secret key

Browse files
Files changed (1) hide show
  1. endpoints.py +3 -2
endpoints.py CHANGED
@@ -4,7 +4,8 @@ from routers import inference, training
4
  from huggingface_hub import login
5
  from config import settings
6
 
7
- login(settings.huggingface_key)
 
8
 
9
  app = FastAPI(openapi_url="/api/v1/sparrow-ml/openapi.json", docs_url="/api/v1/sparrow-ml/docs")
10
 
@@ -22,4 +23,4 @@ app.include_router(training.router, prefix="/api-training/v1/sparrow-ml", tags=[
22
 
23
  @app.get("/")
24
  async def root():
25
- return {"message": "Sparrow ML API"}
 
4
  from huggingface_hub import login
5
  from config import settings
6
 
7
+ # login(settings.huggingface_key)
8
+ login(os.environ.get("HUGGINFACE_KEY"))
9
 
10
  app = FastAPI(openapi_url="/api/v1/sparrow-ml/openapi.json", docs_url="/api/v1/sparrow-ml/docs")
11
 
 
23
 
24
  @app.get("/")
25
  async def root():
26
+ return {"message": "Naivas GRN inferencing"}