Update models.py

models.py

@@ -3,8 +3,8 @@ Model management for Frame 0 Laboratory for MIA
 BAGEL 7B integration via API calls
 """
 
+import spaces
 import logging
-import tempfile
 import os
 from typing import Optional, Dict, Any, Tuple
 from PIL import Image
@@ -88,6 +88,7 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
         except Exception as e:
             logger.warning(f"Failed to cleanup temp file: {e}")
 
+    @spaces.GPU(duration=60)
     def analyze_image(self, image: Image.Image, prompt: str = None) -> Tuple[str, Dict[str, Any]]:
         """Analyze image using BAGEL API"""
         if not self.is_initialized: