Update app.py

app.py

@@ -74,7 +74,7 @@ def server(input, output, session: Session):
 
                 # Run prediction with original BGR image
                 prediction = predictor(im)
-                classes = pred["instances"].pred_classes.tolist()
+                classes = prediction["instances"].pred_classes.tolist()
 
                 results.append({
                     "filename": file["name"],