Update app.py

app.py

@@ -44,8 +44,12 @@ def inference(input_image):
     # Read the categories
     #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()]
+    #with open("dog_cat.txt", "r") as f:
+    categories = []
+    categories.append("cat")
+    categories.append("dog")
+
+    #categories = [s.strip() for s in f.readlines()]
     # Show top categories per image
     top5_prob, top5_catid = torch.topk(probabilities, 2)
     result = {}