Testing with Deepgaze

app.py

@@ -67,14 +67,14 @@ def process_image(
 
     itti_saliency_map, itti_heatmap = detect_and_annotate(
         input_image, GRIDSIZE, 'itti')
-    # deepgaze_saliency_map, deepgaze_heatmap = detect_and_annotate(
-    #     input_image, GRIDSIZE, 'deepgaze')
+    deepgaze_saliency_map, deepgaze_heatmap = detect_and_annotate(
+        input_image, GRIDSIZE, 'deepgaze')
 
     return (
         itti_saliency_map,
         itti_heatmap,
-        # deepgaze_saliency_map,
-        # deepgaze_heatmap,
+        deepgaze_saliency_map,
+        deepgaze_heatmap,
     )
 
 grid_size_Component = gr.Slider(
@@ -101,24 +101,24 @@ with gr.Blocks() as demo:
             type='pil',
             label='Input'
         )
+    with gr.Row():
         itti_saliency_map = gr.Image(
             type='pil',
             label='Itti Saliency Map'
         )
-    with gr.Row():
         itti_heatmap = gr.Image(
             type='pil',
             label='Itti Saliency Ranking Heatmap'
         )
-    # with gr.Row():
-    #     deepgaze_saliency_map = gr.Image(
-    #         type='pil',
-    #         label='DeepGaze Saliency Map'
-    #     )
-    #     deepgaze_heatmap = gr.Image(
-    #         type='pil',
-    #         label='DeepGaze Saliency Ranking Heatmap'
-    #     )
+    with gr.Row():
+        deepgaze_saliency_map = gr.Image(
+            type='pil',
+            label='DeepGaze Saliency Map'
+        )
+        deepgaze_heatmap = gr.Image(
+            type='pil',
+            label='DeepGaze Saliency Ranking Heatmap'
+        )
     submit_button_component = gr.Button(
         value='Submit',
         scale=1,
@@ -134,8 +134,8 @@ with gr.Blocks() as demo:
         outputs=[
             itti_saliency_map,
             itti_heatmap,
-            # deepgaze_saliency_map,
-            # deepgaze_heatmap,
+            deepgaze_saliency_map,
+            deepgaze_heatmap,
         ]
     )
 
@@ -148,8 +148,8 @@ with gr.Blocks() as demo:
         outputs=[
             itti_saliency_map,
             itti_heatmap,
-            # deepgaze_saliency_map,
-            # deepgaze_heatmap,
+            deepgaze_saliency_map,
+            deepgaze_heatmap,
         ]
     )