zerogpu support + gradio fix

app.py

@@ -9,12 +9,27 @@ import requests
 from io import BytesIO
 from pathlib import Path
 
+# Load .env for local development (ignored if not installed / not present)
+try:
+    from dotenv import load_dotenv
+    load_dotenv()
+except ImportError:
+    pass
+
+# Compatibility shim — newer huggingface_hub removed HfFolder which old gradio needs
+import huggingface_hub as _hfhub
+if not hasattr(_hfhub, "HfFolder"):
+    class _HfFolder:
+        @staticmethod
+        def get_token(): return _hfhub.get_token()
+        @staticmethod
+        def save_token(token): pass
+    _hfhub.HfFolder = _HfFolder
+
+import spaces
 import gradio as gr
 from PIL import Image
 from huggingface_hub import hf_hub_download
-from dotenv import load_dotenv
-
-load_dotenv()
 
 # ---------------------------------------------------------------------------
 # Paths — use /data (persistent storage) if available, else /home/user
@@ -157,6 +172,7 @@ def setup():
 # ---------------------------------------------------------------------------
 # Inference
 # ---------------------------------------------------------------------------
+@spaces.GPU(duration=180)
 def generate(target_img, clothing_img, progress=gr.Progress(track_tqdm=True)):
     if target_img is None or clothing_img is None:
         raise gr.Error("Please upload both images before generating.")
@@ -281,7 +297,7 @@ with gr.Blocks(title="Cloth Swap", theme=gr.themes.Soft()) as demo:
             clothing_input = gr.Image(label="Clothing reference", type="numpy", height=400)
         btn    = gr.Button("✨ Generate", variant="primary", size="lg")
         output = gr.Image(label="Result", height=500)
-        gr.Markdown("*Generation takes ~1–2 minutes on T4 GPU.*")
+        gr.Markdown("*Generation takes ~1–2 minutes.*")
 
         btn.click(
             fn=generate,

requirements.txt

@@ -1,2 +1,3 @@
 requests
 Pillow
+python-dotenv