Update app.py

app.py

@@ -23,8 +23,9 @@ def predict(image):
 
     heatmap, pred_id = compute_heatmap(model, image_tensor)
     overlay, _ = upsampleHeatmap(heatmap, image_tensor)
-
-    return overlay, f"Predicted class: {pred_id}"
+    labels = open('imagenet_classes.txt')
+    pred_label = labels[pred_id]
+    return overlay, f"Predicted class: {pred_label}"
 
 
 demo = gr.Interface(