Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 
 import torch
@@ -77,6 +78,7 @@ def visualize_all(pil_image, heatmaps, bboxes, inout_scores, inout_thresh=0.5):
 
     return overlay_image
 
+@spaces.GPU() # ZeroGPU ready
 def main(image_input, progress=gr.Progress(track_tqdm=True)):
     # load image
     image = Image.open(image_input)
@@ -148,6 +150,10 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 input_image = gr.Image(label="Image Input", type="filepath")
                 submit_button = gr.Button("Submit")
+                gr.Examples(
+                    examples = ["examples/the_office.png", "examples/succession.png"]
+                    inputs = [input_image]
+                )
             with gr.Column():
                 result = gr.Image(label="Result", columns=3)
                 heatmaps = gr.Gallery(label="Heatmap")