Update src/enhanced_ai_processor.py

src/enhanced_ai_processor.py

@@ -32,6 +32,7 @@ class EnhancedAIProcessor:
         self.model_version = "v1.2.0"  # Version for tracking
         self._initialize_models()
 
+    @spaces.GPU(enable_queue=True, duration=90) 
     def _initialize_models(self):
         """Initialize all AI models including real-time models"""
         try:
@@ -402,7 +403,8 @@ class EnhancedAIProcessor:
         except Exception as e:
             logging.error(f"Guidelines query error: {e}")
             return f"Error querying guidelines: {str(e)}"
-
+            
+    @spaces.GPU(enable_queue=True, duration=90) 
     def generate_final_report(self, patient_info: Dict[str, Any], visual_results: Dict[str, Any], 
                             guideline_context: str, image_pil: Image.Image, max_new_tokens: int = None) -> str:
         """Generate comprehensive medical report using MedGemma with enhanced tracking"""