Update tongue_model.py

tongue_model.py

@@ -53,10 +53,11 @@ class TongueArcResNet(nn.Module):
 
 # --- 2. 定義預處理與推論類別 ---
 class TongueModelWrapper:
-    def __init__(self, model_path, num_classes=2):
+    def __init__(self, model_path, num_classes=3):
         self.device = torch.device("cpu")
         self.model = TongueArcResNet(num_classes=num_classes)
-        self.model.load_state_dict(torch.load(model_path, map_location=self.device))
+        torch.serialization.add_safe_globals([np._core.multiarray.scalar])
+        self.model.load_state_dict(torch.load(model_path, map_location=self.device, weights_only=False))
         self.model.eval()
         self.transform = transforms.Compose([
             transforms.ToTensor(),