Update app.py

app.py

@@ -46,7 +46,14 @@ from diffusers import DiffusionPipeline
 import numpy as np
 import random
 
-# Load FLUX.1-schnell model
+from huggingface_hub import login
+import os
+import torch
+from diffusers import DiffusionPipeline
+
+# Authenticate first
+login(token=os.getenv("HUGGINGFACE_HUB_TOKEN"))
+
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -55,6 +62,7 @@ pipe = DiffusionPipeline.from_pretrained(
     torch_dtype=torch_dtype
 ).to(device)
 
+
 MAX_SEED = np.iinfo(np.int32).max