Update app.py

app.py

@@ -32,7 +32,7 @@ predictor = DefaultPredictor(cfg)
 
 
 def inference(img):
-    class_names = ['astro', 'cort', 'sh-sy5y']
+    class_names = ['0','1','2'] #['astro', 'cort', 'sh-sy5y']
     im = np.asarray(Image.open(img).convert('RGB'))
     outputs = predictor(im)
     pred_classes = outputs['instances'].pred_classes.cpu().numpy().tolist()