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

app.py

@@ -15,6 +15,7 @@ import cv2
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 from functools import partial
+import tempfile
 
 class Net2(nn.Module):
     def __init__(self):
@@ -238,11 +239,22 @@ def process_image(input_image, model, threshold=0.5):
     return ship_images, int(elapsed_time)
 
 
-def gradio_process_image(input_image_path, model, threshold=0.5):
+def gradio_process_image(input_image, model, threshold=0.5):
     start_time = time.time()
-    output_image_path = scanmap(input_image_path, model, threshold)
+
+    # save PIL image to a temporary file
+    temp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+    temp.close()
+    input_image.save(temp.name)
+
+    # pass file path to scanmap
+    output_image_path = scanmap(temp.name, model, threshold)
+
     elapsed_time = time.time() - start_time
 
+    # delete temporary file after processing
+    os.unlink(temp.name)
+
     return output_image_path, f"Elapsed Time (seconds): {elapsed_time}"
 
 inputs = gr.inputs.Image(label="Upload Image")