DemoSpace / app.py
revi13's picture
Update app.py
e0bcf49 verified
# app.py
import os
from fastapi import FastAPI
from huggingface_hub import login
app = FastAPI()
@app.get("/")
def root():
login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
return {"status": "ok"}
# app.py
import os
from fastapi import FastAPI
from huggingface_hub import hf_hub_download
# 必ず import の前に環境変数を設定!
os.environ["HF_HOME"] = "/tmp/hf"
os.environ["HF_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
os.environ["XDG_CACHE_HOME"] = "/tmp/hf/xdg"
app = FastAPI()
@app.get("/")
def root():
model_path = hf_hub_download(
repo_id="revi13/ip-adapter-faceid-private",
filename="ip-adapter-faceid-plusv2_sd15.bin",
local_dir="/tmp/hf/models", # 保存先を手動指定
local_dir_use_symlinks=False, # シンボリックリンクではなく実体保存
token=os.environ["HUGGINGFACE_HUB_TOKEN"]
)
return {"model_path": model_path}