Changed scan func

app.py

@@ -194,7 +194,7 @@ else:
 # iface = gr.Interface(fn=gradio_process_image, inputs=inputs, outputs=outputs)
 # iface.launch()
 
-def scanmap(image_path, model, device, threshold=0.5):
+def scanmap(image_path, model, threshold=0.5):
     satellite_image = cv2.imread(image_path)
     satellite_image = satellite_image.astype(np.float32) / 255.0
 
@@ -203,6 +203,8 @@ def scanmap(image_path, model, device, threshold=0.5):
 
     height, width, channels = satellite_image.shape
 
+    # ensure model is in float32 precision
+    model.float()
 
     fig, ax = plt.subplots(1)
     ax.imshow(satellite_image)
@@ -213,7 +215,6 @@ def scanmap(image_path, model, device, threshold=0.5):
         for x in range(0, width - window_size[0] + 1, stride):
             cropped_window = satellite_image[y:y + window_size[1], x:x + window_size[0]]
             cropped_window_torch = torch.tensor(cropped_window.transpose(2, 0, 1), dtype=torch.float32).unsqueeze(0)
-            cropped_window_torch = cropped_window_torch.to(device)  # move data to the same device as model
 
             with torch.no_grad():
                 probabilities = model(cropped_window_torch)
@@ -231,6 +232,7 @@ def scanmap(image_path, model, device, threshold=0.5):
 
     return output_path
 
+
 def process_image(input_image, model, threshold=0.5):
     start_time = time.time()
     ship_images = scanmap(input_image, model, threshold)