Update Colony_Analyzer_AI_zstack2_HF.py

Colony_Analyzer_AI_zstack2_HF.py

@@ -61,7 +61,9 @@ def visualize_segmentation(mask, image=0):
 import torch
 from transformers import SegformerForSemanticSegmentation
 # Load fine-tuned model
-model = SegformerForSemanticSegmentation.from_pretrained("ReyaLabColumbia/Segformer_Colony_Counter")  # Adjust path
+#ReyaLabColumbia/Segformer_Colony_Counter
+#ReyaLabColumbia/Organoid_Counter
+model = SegformerForSemanticSegmentation.from_pretrained("ReyaLabColumbia/Organoid_Counter")  # Adjust path
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 model.eval()  # Set to evaluation mode