Update app.py

app.py

@@ -132,14 +132,25 @@ def update_button(filepath, z):
     print(filepath_show)
     return filepath_show, [filepath], fp0, fp0
 
-def detect_cells(filepath):
+def detect_cells(filepath, z):
     model = tf.keras.models.load_model('/model_positions', compile=False)
 
     img = tiff.imread(filepath)
     img = img/np.max(img)
 
+    tensor = tf.convert_to_tensor(img)
+
+    result = model(tensor)
+    print(filepath)
+    fpath, fext = os.path.splitext(filepath)
+
+    filepath_result = fpath+'result'+'.tiff'
     
-    return fp1, fp1
+    tiff.imwrite(filepath_result, result)
+
+    filepath_result_show = tif_view(filepath_result, z)
+
+    return filepath_result, fp1, fp1
 
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -190,7 +201,7 @@ with gr.Blocks(title = "Hello",
     #depth.change(update_depth, [up_btn, depth], depth)
 
     
-    send_btn.click(detect_cells, [up_btn], [output_image1, output_image2])# flows, down_btn, down_btn2])
+    send_btn.click(detect_cells, [up_btn], [down_btn, output_image1, output_image2])# flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])