JointTaggerProject-Inference-Beta-AttnVis

Sleeping

drhead commited on May 1

Commit

a6b4c50

verified ·

1 Parent(s): f65a662

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -239,7 +239,8 @@ def create_cam_visualization_pil(image_pil, cam, alpha=0.6, vis_threshold=0.2):
     Returns:
         PIL.Image.Image with overlay
     """
     w, h = image_pil.size
     # Resize CAM to match image

     Returns:
         PIL.Image.Image with overlay
     """
+    if cam is None:
+        return image_pil
     w, h = image_pil.size
     # Resize CAM to match image