Update app.py

app.py

@@ -33,6 +33,21 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# Try to import spaces for ZeroGPU support (with automatic CPU fallback)
+try:
+    import spaces
+    ZEROGPU_AVAILABLE = True
+    logger.info("✅ ZeroGPU support enabled - GPU acceleration available")
+except ImportError:
+    ZEROGPU_AVAILABLE = False
+    logger.info("ℹ️ ZeroGPU not available - will use CPU (slower but functional)")
+    # Create a dummy decorator for compatibility
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            """Dummy decorator when ZeroGPU is not available"""
+            return func
+
 # Import application modules
 try:
     from main import process_video
@@ -81,6 +96,7 @@ def validate_video_file(video_path: str) -> Tuple[bool, str]:
         return False, f"Error validating video: {str(e)}"
 
 
+@spaces.GPU  # Enable ZeroGPU acceleration (automatic CPU fallback if unavailable)
 def estimate_vehicle_speed(
     video_file: str,
     model_choice: str,
@@ -91,6 +107,10 @@ def estimate_vehicle_speed(
     """
     Process video and estimate vehicle speeds.
     
+    This function is decorated with @spaces.GPU to enable automatic GPU
+    acceleration when running on ZeroGPU Spaces. If GPU is not available,
+    it automatically falls back to CPU processing.
+    
     Args:
         video_file: Path to uploaded video
         model_choice: YOLO model selection