Update app.py

app.py

@@ -7,6 +7,9 @@ from PIL import Image
 import tensorflow as tf
 from skimage.feature import peak_local_max
 
+model_adresses = {"Intestinal organoids (0.32x0.32x2.0 um)": './model_positions',
+                  "c. Elegans embryo (0.1x0.1x1.0 um)": './model_positions'}
+
 
 fp0 = np.zeros((96, 128), dtype = np.uint8)
 
@@ -191,8 +194,10 @@ def update_z(filepath, filepath_result, filepath_coordinates, z):
     
     return (filepath_show, display_boxes), (filepath_result_show, display_boxes)
 
-def detect_cells(filepath, z):
-    model = tf.keras.models.load_model('./model_positions', compile=False)
+def detect_cells(filepath, z, model):
+
+
+    model = tf.keras.models.load_model(model_adresses[model], compile=False)
 
     img = tiff.imread(filepath[-1])
     
@@ -324,10 +329,14 @@ with gr.Blocks(title = "Hello",
             with gr.Row():
                 with gr.Column(scale=1):                    
                     with gr.Row():
-                        depth = gr.Number(label = 'z-depth', value = 10)
+                        depth = gr.Slider(0, 100, step=1, label = 'z-depth', value = 10)
                         channel = gr.Number(label = 'channel (does nothing)', value = 0)
                         scale = gr.Number(label = 'resize (does nothing)', value = 0)
 
+                    model = gr.Dropdown(
+                            model_adresses.keys(), label="Detection model (with resolutions)", info="Will add more models later!"
+                                )
+                    
                     up_btn = gr.UploadButton("Upload image volume (tif/tiff)", visible=True,  file_count = "multiple")                        
 
                     #gr.HTML("""<h4 style="color:white;"> Note2: Only the first image of a tif will display the segmentations, but you can download segmentations for all planes. </h4>""")
@@ -365,7 +374,7 @@ with gr.Blocks(title = "Hello",
     #depth.change(update_depth, [up_btn, depth], depth)
 
     # DO NOT RENDER OUTPUT TWICE
-    send_btn.click(detect_cells, [up_btn, depth], [down_btn, down_btn2, output_image]).then(update_z,  [up_btn, down_btn, down_btn2, depth], [input_image, output_image])# flows, down_btn, down_btn2])
+    send_btn.click(detect_cells, [up_btn, depth, model], [down_btn, down_btn2, output_image]).then(update_z,  [up_btn, down_btn, down_btn2, depth], [input_image, output_image])# flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])