vggt_da

app.py

@@ -15,6 +15,12 @@ import zlib
 from pathlib import Path
 from einops import rearrange
 from typing import List, Tuple, Union
+try:
+    import spaces   
+except ImportError:
+    # Fallback for local development
+    def spaces(func):
+        return func
 import torch
 import logging
 from concurrent.futures import ThreadPoolExecutor
@@ -33,13 +39,6 @@ except ImportError as e:
     logger.error(f"Failed to import custom modules: {e}")
     raise
 
-try:
-    import spaces   
-except ImportError:
-    # Fallback for local development
-    def spaces(func):
-        return func
-
 # Constants
 MAX_FRAMES = 80
 COLORS = [(0, 0, 255), (0, 255, 255)]  # BGR: Red for negative, Yellow for positive