Update app.py

app.py

@@ -234,7 +234,7 @@ def update(filepath, filepath_result, filepath_coordinates, z, c):
     
     return (filepath_show, display_boxes), (filepath_result_show, display_boxes)
 
-def detect_cells(filepath, c, model, rescale_z, rescale_xy):
+def detect_cells(filepath, c, model, rescale_z, rescale_xy, progress=gr.Progress()):
 
     model = tf.keras.models.load_model(model_adresses[model], compile=False)
 
@@ -281,7 +281,7 @@ def detect_cells(filepath, c, model, rescale_z, rescale_xy):
 
     #display_boxes = filter_coordinates(filepath_coordinates, z)
 
-    return filepath_result, filepath_coordinates#, (filepath_result_show, display_boxes)
+    return filepath_result, filepath_coordinates, (fp0, [((5, 5, 10, 10), 'nothing')])#, (filepath_result_show, display_boxes)
 
 def pad(img, z_tile = 32, z_buffer=2, xy_tile = 32):
 
@@ -424,7 +424,7 @@ with gr.Blocks(title = "Hello",
     #depth.change(update_depth, [up_btn, depth], depth)
 
     # Prediction
-    send_btn.click(detect_cells, [up_btn, channel, model, rescale_z, rescale_xy], [down_btn, down_btn2]).then(update,  [up_btn, down_btn, down_btn2, depth, channel], [input_image, output_image])# flows, down_btn, down_btn2])
+    send_btn.click(detect_cells, [up_btn, channel, model, rescale_z, rescale_xy], [ down_btn, down_btn2, output_image]).then(update,  [up_btn, down_btn, down_btn2, depth, channel], [input_image, output_image])# flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])