Update app.py

app.py

@@ -88,7 +88,22 @@ class ModelManager:
                 'ConvNextV2Base_best_model.pth',
                 map_location=self.device
             )
-            self._breed_model.load_state_dict(checkpoint['base_model'], strict=False)
+            if 'base_model' in checkpoint:
+                
+                self._breed_model.load_state_dict(checkpoint['base_model'], strict=False)
+            elif 'model' in checkpoint:
+                
+                self._breed_model.load_state_dict(checkpoint['model'], strict=False)
+            elif 'state_dict' in checkpoint:
+               
+                self._breed_model.load_state_dict(checkpoint['state_dict'], strict=False)
+            else:
+                
+                try:
+                    self._breed_model.load_state_dict(checkpoint, strict=False)
+                except Exception as e:
+                    print(f"嘗試載入模型權重時出錯: {e}")
+                    print(f"checkpoint鍵: {list(checkpoint.keys()) if isinstance(checkpoint, dict) else 'Not a dictionary'}")                    
             self._breed_model.eval()
         return self._breed_model