Spaces:

DawnC
/

PawMatchAI

Running on Zero

DawnC commited on Dec 3, 2024

Commit

d428527

1 Parent(s): 6526fb2

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(