Update app.py

app.py

@@ -593,8 +593,8 @@ class ModelManager:
     _initialized = False
     
     def __init__(self):
-        self.model_yolo = None  # YOLO模型
-        self.model = None       # 品種分類模型
+        self.model_yolo = None
+        self.model = None
     
     @classmethod
     async def get_instance(cls):
@@ -602,20 +602,14 @@ class ModelManager:
             cls._instance = cls()
         return cls._instance
     
-    @spaces.GPU
     async def initialize(self):
-        # 只初始化一次模型
         if not self._initialized:
-            # 初始化 YOLO
             self.model_yolo = YOLO('yolov8l.pt')
-            
-            # 初始化品種分類模型
             num_classes = len(dog_breeds)
             self.model = BaseModel(num_classes=num_classes)
             checkpoint = torch.load('124_best_model_dog.pth')
             self.model.load_state_dict(checkpoint['base_model'], strict=False)
             self.model.eval()
-            
             self._initialized = True