Changed scan func

.idea/workspace.xml

@@ -65,7 +65,7 @@
       <workItem from="1683665300392" duration="7649000" />
       <workItem from="1683708398011" duration="1235000" />
       <workItem from="1684437905081" duration="110000" />
-      <workItem from="1686602174110" duration="3352000" />
+      <workItem from="1686602174110" duration="3739000" />
     </task>
     <servers />
   </component>

app.py

@@ -234,7 +234,13 @@ def process_image(input_image, model, threshold=0.5):
 def gradio_process_image(input_image, model, threshold=0.5):
     ship_images, elapsed_time = process_image(input_image, model, threshold)
 
-    return ship_images, f"Elapsed Time (seconds): {elapsed_time}"
+    if len(ship_images) > 0:
+        # Convert first image to format compatible with Gradio
+        output_image = Image.fromarray((ship_images[0] * 255).astype(np.uint8))
+    else:
+        output_image = None
+
+    return output_image, f"Elapsed Time (seconds): {elapsed_time}"
 
 
 inputs = gr.Image(label="Upload Image")