Update app.py

app.py

@@ -53,7 +53,7 @@ def inference(input_img, transparancy = 0.5, target_layer_number = -1):
     visualization = show_cam_on_image(org_img/255,grayscale_cam,use_rgb=True,
                                      image_weight=transparancy)
     
-    return classes[prediction[0].item()], visualization, confidences
+    return classes[int(prediction[0].item())], visualization, confidences
     
 demo = gr.Interface(
     inference,