zero gpu

diff_ai.py

@@ -29,6 +29,11 @@ All models are lazy-loaded (thread-safely). Falls back to classical pipeline
 (diff.py) if any required model is unavailable — see app.py for the wiring.
 """
 import threading
+try:
+    import spaces
+    HAS_SPACES = True
+except ImportError:
+    HAS_SPACES = False
 
 import cv2
 import numpy as np
@@ -453,7 +458,7 @@ def _composite_object(base, object_img, mask, bbox, color):
 # MAIN ENTRY
 # =====================================================================
 
-def compare_images_ai(img1_path, img2_path, output_path, heatmap_path):
+def _compare_images_ai_impl(img1_path, img2_path, output_path, heatmap_path):
     """End-to-end change-region-driven AI pipeline. Raises RuntimeError
     if any required model is unavailable."""
     if not _get_loftr() or not _get_sam() or not _get_dinov2():
@@ -637,4 +642,13 @@ def compare_images_ai(img1_path, img2_path, output_path, heatmap_path):
         "object_changes": n_total,
         "severity": severity,
         "alignment_failed": alignment_failed,
-    }
+    }
+
+
+if HAS_SPACES:
+    @spaces.GPU
+    def compare_images_ai(img1_path, img2_path, output_path, heatmap_path):
+        return _compare_images_ai_impl(img1_path, img2_path, output_path, heatmap_path)
+else:
+    def compare_images_ai(img1_path, img2_path, output_path, heatmap_path):
+        return _compare_images_ai_impl(img1_path, img2_path, output_path, heatmap_path)