Changes on scan

app.py

@@ -160,6 +160,12 @@ def scanmap(satellite_image, model):
 
 
 def gradio_process_image(input_image):
+    if input_image is None:
+        return None, None
+
+    if isinstance(input_image, np.ndarray):
+        input_image = Image.fromarray(input_image)
+
     image_np = np.array(input_image.convert("RGB"))
     heatmap = scanmap(image_np, model)
     heatmap_normalized = (heatmap - np.min(heatmap)) / (np.max(heatmap) - np.min(heatmap))
@@ -174,4 +180,4 @@ outputs = [
 ]
 
 iface = gr.Interface(fn=gradio_process_image, inputs=inputs, outputs=outputs)
-iface.launch()
+iface.launch(share=True)