change dockerfile

Dockerfile

@@ -2,6 +2,12 @@ FROM python:3.10
 
 WORKDIR /app
 
+ENV HF_HOME=/app/hf_cache
+ENV TRANSFORMERS_CACHE=/app/hf_cache
+ENV DIFFUSERS_CACHE=/app/hf_cache
+
+RUN mkdir -p /app/hf_cache
+
 RUN pip install --no-cache-dir torch torchvision torchaudio diffusers fastapi uvicorn
 RUN pip install git+https://github.com/huggingface/diffusers.git
 RUN pip install transformers

app.py

@@ -4,12 +4,12 @@ import torch
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
 from diffusers import FluxPipeline  # type: ignore
+from huggingface_hub import login
 
 app = FastAPI()
 
 # Récupération du token et authentification
 token = os.getenv("HF_TOKEN")
-
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload()