Update src/ai_processor.py

src/ai_processor.py

@@ -16,7 +16,7 @@ import cv2
 import numpy as np
 from PIL import Image
 from PIL.ExifTags import TAGS
-
+import spaces
 # --- Logging config ---
 logging.basicConfig(
     level=getattr(logging, LOGLEVEL, logging.INFO),
@@ -26,12 +26,6 @@ logging.basicConfig(
 def _log_kv(prefix: str, kv: Dict):
     logging.debug(prefix + " | " + " | ".join(f"{k}={v}" for k, v in kv.items()))
 
-# --- Spaces GPU decorator (REQUIRED) ---
-from spaces import GPU as _SPACES_GPU
-
-@_SPACES_GPU(enable_queue=True)
-def smartheal_gpu_stub(ping: int = 0) -> str:
-    return "ready"
 
 # ---- Paths / constants ----
 UPLOADS_DIR = "uploads"
@@ -918,7 +912,7 @@ Automated analysis provides quantitative measurements; verify via clinical exami
             logging.error(f"Failed to save/commit image: {e}")
             return ""
             
-    @_SPACES_GPU(enable_queue=True)
+    @spaces.GPU
     def full_analysis_pipeline(self, image_pil: Image.Image, questionnaire_data: Dict) -> Dict:
         try:
             saved_path = self.save_and_commit_image(image_pil)
@@ -964,7 +958,7 @@ Automated analysis provides quantitative measurements; verify via clinical exami
                 "saved_image_path": None,
                 "guideline_context": "",
             }
-    @_SPACES_GPU(enable_queue=True)
+    @spaces.GPU
     def analyze_wound(self, image, questionnaire_data: Dict) -> Dict:
         try:
             if isinstance(image, str):