Update app.py

app.py

@@ -1,13 +1,34 @@
 import os
-# PyTorch 2.8 (temporary hack)
-os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
+import sys
+from pathlib import Path
 
-# Fix ZeroGPU offload directory issue
-zerogpu_offload_dir = os.environ.get('ZEROGPU_OFFLOAD_DIR', '/tmp/zerogpu-offload')
+# Fix 1: Ensure the offload directory exists with proper permissions
+zerogpu_offload_dir = '/data-nvme/zerogpu-offload'
 if not os.path.exists(zerogpu_offload_dir):
-    os.makedirs(zerogpu_offload_dir, exist_ok=True)
+    try:
+        os.makedirs(zerogpu_offload_dir, exist_ok=True, mode=0o755)
+    except Exception as e:
+        # Fallback to tmp directory
+        zerogpu_offload_dir = '/tmp/zerogpu-offload'
+        os.makedirs(zerogpu_offload_dir, exist_ok=True, mode=0o755)
+
+# Fix 2: Set environment variable before spaces import
 os.environ['ZEROGPU_OFFLOAD_DIR'] = zerogpu_offload_dir
 
+# Fix 3: Patch the pathlib.Path.mkdir method to handle FileExistsError
+original_mkdir = Path.mkdir
+
+def patched_mkdir(self, mode=0o777, parents=False, exist_ok=False):
+    """Patched mkdir that always uses exist_ok=True for ZeroGPU directories"""
+    if 'zerogpu-offload' in str(self):
+        exist_ok = True
+    return original_mkdir(self, mode=mode, parents=parents, exist_ok=exist_ok)
+
+# Apply the patch
+Path.mkdir = patched_mkdir
+
+print("🔧 Applied ZeroGPU fixes")
+
 
 # --- 1. Model Download and Setup (Diffusers Backend) ---
 import spaces