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="2380000" />
+      <workItem from="1686602174110" duration="2679000" />
     </task>
     <servers />
   </component>

app.py

@@ -14,6 +14,7 @@ import torch
 import cv2
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
+from functools import partial
 
 class Net2(nn.Module):
     def __init__(self):
@@ -248,8 +249,11 @@ outputs = [
     gr.Image(label="Detected Ships"),
     gr.Textbox(label="Elapsed Time")
 ]
+# Here I'm using 0.5 as the threshold, but adjust according to your needs
+gradio_process_image_partial = partial(gradio_process_image, model=model, device=device, threshold=0.5)
 
-iface = gr.Interface(fn=gradio_process_image, inputs=inputs, outputs=outputs)
+iface = gr.Interface(fn=gradio_process_image_partial, inputs=inputs, outputs=outputs)
 iface.launch()
 
 
+