Spaces:

DawnC
/

PawMatchAI

Running on Zero

DawnC commited on about 22 hours ago

Commit

bf35501

•

1 Parent(s): 761ce54

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -175,7 +175,7 @@ class ModelManager:
         if self._breed_model is None:
             self._breed_model = BaseModel(
                 num_classes=len(dog_breeds),
-                device=self.device  # 使用我們的device屬性
             ).to(self.device)
             checkpoint = torch.load(

         if self._breed_model is None:
             self._breed_model = BaseModel(
                 num_classes=len(dog_breeds),
+                device=self.device
             ).to(self.device)
             checkpoint = torch.load(