[martin-dev] fix gpu handle

demo/launch_gradio.py

@@ -441,6 +441,7 @@ def get_module_similarity_pooled(
         hook_handle.remove()
 
 
+@GPU(duration=60)
 def process_dual_inputs(
     model_choice: str,
     selected_layer: str,
@@ -546,7 +547,6 @@ def process_dual_inputs(
         return None, f'Error: {str(e)}'
 
 
-@GPU
 def create_demo() -> gr.Blocks:
     """Create and configure the Gradio demo interface for dual image comparison.
 
@@ -635,21 +635,13 @@ def create_demo() -> gr.Blocks:
             outputs=[plot_output, info_output]
         )
 
-        # Custom interactive examples (safe for GPU Spaces)
-        # gr.Markdown("### Example Prompts (Click to Insert)")
-        # with gr.Row():
-        #     gr.Button("Example 1").click(
-        #         fn=lambda: ("What is in this image? Describe in one word.", None, None),
-        #         outputs=[instruction_input, image1_input, image2_input]
-        #     )
-        #     gr.Button("Example 2").click(
-        #         fn=lambda: ("Describe the main object in the picture in one word.", None, None),
-        #         outputs=[instruction_input, image1_input, image2_input]
-        #     )
-        #     gr.Button("Example 3").click(
-        #         fn=lambda: ("What color is the dominant object? Describe in one word.", None, None),
-        #         outputs=[instruction_input, image1_input, image2_input]
-        #     )
+        # Add examples
+        gr.Examples(
+            examples=[
+                [example1()], [example2()], [example3()]
+            ],
+            inputs=[instruction_input, image1_input, image2_input]
+        )
             
     return demo