Upload 2 files

Colony_Analyzer_AI2_HF.py

@@ -258,7 +258,6 @@ def main(args):
     img_map = cut_img(args[0])
     for z in img_map:
         mask = eval_img(img_map[z])
-        cv2.imwrite(img_map[z], mask)
     del mask,z
     p = stitch(img_map)
     colonies = analyze_colonies(p, min_size, min_circ)
@@ -298,7 +297,7 @@ def main(args):
 
 
 
-    #cv2.imwrite('results.png', np.vstack((img, caption)))
+    cv2.imwrite('results.png', np.vstack((img, caption)))
 
-    return(np.vstack((img, caption)))
+    return(np.vstack((img, caption)), 'results.png', 'results.xlsx')
 

app.py

@@ -25,7 +25,11 @@ iface = gr.Interface(
         gr.Number(label="Minimum Colony Size (pixels)", value=1000),
         gr.Number(label="Minimum Circularity", value=0.25)
     ],
-    outputs=[gr.Image(type="pil", label="Analyzed Image"), gr.File(label="Download results (Excel)")],
+    outputs=[
+        gr.Image(type="pil", label="Analyzed Image"),
+        gr.File(label="Download Image"),
+        gr.File(label="Download results (Excel)")
+    ],
     title="AI Colony Analyzer",
     description="Upload an image to run the colony analysis."
 )