added login

.gitignore

@@ -0,0 +1 @@
+.env

Dockerfile

@@ -8,6 +8,7 @@ RUN pip install git+https://github.com/huggingface/diffusers.git
 RUN pip install transformers
 RUN pip install accelerate
 RUN pip install huggingface_hub
+RUN pip install python-dotenv
 
 COPY . /app
 

app.py

@@ -1,11 +1,14 @@
+import os
 import torch
 from diffusers import FluxPipeline # type: ignore
 import gradio as gr # type: ignore
 from huggingface_hub import login
-login()
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
+token = os.getenv("HF_TOKEN")
+login(token=token)
+
 
 prompt = "A cat holding a sign that says hello world"
 image = pipe(