Changed scan func

app.py

@@ -249,6 +249,8 @@ outputs = [
     gr.Image(label="Detected Ships"),
     gr.Textbox(label="Elapsed Time")
 ]
+model = None  # TODO: initialize your model
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Here I'm using 0.5 as the threshold, but adjust according to your needs
 gradio_process_image_partial = partial(gradio_process_image, model=model, device=device, threshold=0.5)