Update app.py

app.py

@@ -23,7 +23,9 @@ def predict(image):
 
     heatmap, pred_id = compute_heatmap(model, image_tensor)
     overlay, _ = upsampleHeatmap(heatmap, image_tensor)
-    labels = open('imagenet_classes.txt')
+    with open("imagenet_classes.txt", "r") as f:
+    labels = f.read().splitlines()
+
     pred_label = labels[pred_id]
     return overlay, f"Predicted class: {pred_label}"