Changed scan func

app.py

@@ -213,15 +213,19 @@ def scanmap(image_np, model, threshold=0.5):
 
             # if probability is greater than threshold, draw a bounding box
             if probabilities[0, 1].item() > threshold:
-                rect = patches.Rectangle((x, y), window_size[0], window_size[1], linewidth=1, edgecolor='r', facecolor='none')
+                rect = patches.Rectangle((x, y), window_size[0], window_size[1], linewidth=1, edgecolor='r',
+                                         facecolor='none')
                 ax.add_patch(rect)
 
-    # Save the image to a byte buffer
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
+    # Convert matplotlib figure to PIL Image
+    fig.canvas.draw()
+    img_arr = np.array(fig.canvas.renderer.buffer_rgba())
+    plt.close(fig)
+
+    img = Image.fromarray(img_arr)
+
+    return img
 
-    return Image.open(buf)  # return PIL Image
 
 def process_image(input_image):
     image = Image.fromarray(input_image).convert("RGB")
@@ -244,6 +248,5 @@ outputs = [
 ]
 
 iface = gr.Interface(fn=gradio_process_image, inputs=inputs, outputs=outputs)
-iface.launch(launch_mode='external')  # or 'inline'
-
+iface.launch()