|
|
|
|
|
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"} |
|
|
|
|
|
|
|
|
import os |
|
|
from fastapi import FastAPI |
|
|
from huggingface_hub import hf_hub_download |
|
|
|
|
|
|
|
|
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} |
|
|
|