Upload folder using huggingface_hub

app.py

@@ -7,11 +7,19 @@ while preserving image quality. Powered by ZeroGPU.
 Based on: https://github.com/XuandongZhao/WatermarkAttacker
 """
 
+import os
 import gradio as gr
 import numpy as np
 import spaces
 import torch
 from PIL import Image
+from huggingface_hub import login
+
+# Login with HF_TOKEN for gated model access
+if os.environ.get("HF_TOKEN"):
+    login(token=os.environ["HF_TOKEN"])
+    print("Logged in with HF_TOKEN")
+
 from diffusers import StableDiffusion3Img2ImgPipeline
 
 # Model configuration