Update app.py

app.py

@@ -219,7 +219,7 @@ def make_patch_heatmap(img, size=32, stride=16):
             examples.append(create_tf_example(occluded))
             coords.append((y, x))
 
-    # 🔥 batch inference
+
     tensor = tf.constant(examples)
     outputs = infer(inputs=tensor)["embedding"].numpy()
 
@@ -398,13 +398,13 @@ def gradio_predict(image):
 
     #heatmap = make_patch_heatmap(image)
     # stage 1
-    coarse_map = make_patch_heatmap(img)
+    coarse_map = make_patch_heatmap(image)
     
     # หา region ที่สำคัญ
     mask = coarse_map > np.mean(coarse_map)
     
     # stage 2 เฉพาะ mask
-    refined_map = refine_heatmap(img, mask)
+    refined_map = refine_heatmap(image, mask)
     overlay = overlay_heatmap(image, refined_map)
 
     return result, overlay, "✅ Done"