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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py CHANGED
@@ -9,3 +9,26 @@ app = FastAPI()
9
  def root():
10
  login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
11
  return {"status": "ok"}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  def root():
10
  login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
11
  return {"status": "ok"}
12
+
13
+ # app.py
14
+ import os
15
+ from fastapi import FastAPI
16
+ from huggingface_hub import hf_hub_download
17
+
18
+ # 必ず import の前に環境変数を設定!
19
+ os.environ["HF_HOME"] = "/tmp/hf"
20
+ os.environ["HF_HUB_CACHE"] = "/tmp/hf/hub"
21
+ os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
22
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
23
+ os.environ["XDG_CACHE_HOME"] = "/tmp/hf/xdg"
24
+
25
+ app = FastAPI()
26
+
27
+ @app.get("/")
28
+ def root():
29
+ model_path = hf_hub_download(
30
+ repo_id="revi13/ip-adapter-faceid-private",
31
+ filename="ip-adapter-faceid-plusv2_sd15.bin",
32
+ token=os.environ["HUGGINGFACE_HUB_TOKEN"]
33
+ )
34
+ return {"model_path": model_path}