revi13 commited on
Commit
0b92c2a
·
verified ·
1 Parent(s): f113475

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -3,16 +3,9 @@ import os
3
  from fastapi import FastAPI
4
  from huggingface_hub import login
5
 
6
- # login前に cacheディレクトリを作成
7
- os.environ["HF_HOME"] = "/tmp/hf"
8
- os.environ["HF_HUB_CACHE"] = "/tmp/hf/hub"
9
- os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
10
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
11
- os.environ["XDG_CACHE_HOME"] = "/tmp/hf/xdg"
12
-
13
  app = FastAPI()
14
 
15
  @app.get("/")
16
  def root():
17
- login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""), add_to_git_credential=False)
18
  return {"status": "ok"}
 
3
  from fastapi import FastAPI
4
  from huggingface_hub import login
5
 
 
 
 
 
 
 
 
6
  app = FastAPI()
7
 
8
  @app.get("/")
9
  def root():
10
+ login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
11
  return {"status": "ok"}