Update app.py

app.py

@@ -42,7 +42,7 @@ def inference(input_image):
     probabilities = torch.nn.functional.softmax(output[0], dim=0)
 
     # Read the categories
-    with open("imagenet_classes.txt", "r") as f:
+    with open("dog_cat.txt", "r") as f:
         categories = [s.strip() for s in f.readlines()]
     with open("dog_cat.txt", "r") as f:
         categories = [s.strip() for s in f.readlines()]