Changed scan func

app.py

@@ -242,10 +242,10 @@ def process_image(input_image, model, threshold=0.5):
 def gradio_process_image(input_image, model, threshold=0.5):
     start_time = time.time()
 
-    # save PIL image to a temporary file
+    # save numpy array to a temporary file
     temp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
     temp.close()
-    input_image.save(temp.name)
+    cv2.imwrite(temp.name, cv2.cvtColor(input_image * 255, cv2.COLOR_RGB2BGR))
 
     # pass file path to scanmap
     output_image_path = scanmap(temp.name, model, threshold)