login

app.py

@@ -4,13 +4,18 @@ 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")
+token = os.getenv("HF_TOKEN")
 login(token=token)
 
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.enable_model_cpu_offload()
+
 def generate_image(prompt: str):
     image = pipe(
         prompt,