fix bugs

app.py

@@ -16,7 +16,7 @@ def cat_dog_predict(model: torch.nn.Module, img: Image) -> str:
     ])
     img = transform(img)
     img = img.to(device)
-    class_names = ["cat", "dog"]
+    class_names = ["dog", "cat"]
     pred = model(img.unsqueeze(0))
     pred_prob = torch.softmax(pred, 1)
     pred_label = torch.max(pred_prob, 1)