Update Colony_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py

@@ -114,7 +114,7 @@ from transformers import SegformerForSemanticSegmentation
 # Load fine-tuned model
 #ReyaLabColumbia/Segformer_Colony_Counter
 #ReyaLabColumbia/OrganoidCounter
-model = SegformerForSemanticSegmentation.from_pretrained("ReyaLabColumbia/Segformer_Colony_Counter_GP")
+model = SegformerForSemanticSegmentation.from_pretrained("ReyaLabColumbia/Segformer_Organoid_Counter_GP")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 model.eval()  # Set to evaluation mode