Spaces:

sanjanatule
/

ERA_Assignment13

Sleeping

sanjanatule commited on Aug 16, 2023

Commit

efe8778

1 Parent(s): 7885866

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ with gr.Blocks() as demo:
         cam = EigenCAM(inference_model, target_layers, use_cuda=False,reshape_transform=yolov3_reshape_transform)
         grayscale_cam = cam(input_tensor = input_img_tensor, targets= targets)
         grayscale_cam = grayscale_cam[0, :]
-        visualization = show_cam_on_image(input_img_copy, grayscale_cam, use_rgb=True, image_weight=gradcam_opa)
         return (visualization,sections)
       else:

         cam = EigenCAM(inference_model, target_layers, use_cuda=False,reshape_transform=yolov3_reshape_transform)
         grayscale_cam = cam(input_tensor = input_img_tensor, targets= targets)
         grayscale_cam = grayscale_cam[0, :]
+        visualization = show_cam_on_image(input_img_copy, grayscale_cam, use_rgb=False, image_weight=gradcam_opa)
         return (visualization,sections)
       else: