DemoSpace / app.py
revi13's picture
Update app.py
e8b4369 verified
raw
history blame
299 Bytes
# app.py
import os
from fastapi import FastAPI
from huggingface_hub import login
os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/.cache"
os.environ["HF_HOME"] = "/tmp/.cache"
app = FastAPI()
@app.get("/")
def root():
login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
return {"status": "ok"}