Update app.py

app.py

@@ -47,6 +47,16 @@ def inference(input_image):
     # Read the categories
     with open("artist_classes.txt", "r") as f:
         categories = [s.strip() for s in f.readlines()]
+        
+    categories = {
+    0:vanGogh,
+    1:Monet,
+    2:Leonardo da Vinci,
+    3:Rembrandt,
+    4:Pablo Picasso,
+    5:Salvador Dali
+    }
+    
     # Show top categories per image
     top5_prob, top5_catid = torch.topk(probabilities, 6)
     result = {}