Update README.md

README.md

@@ -33,7 +33,7 @@ from transformers import AutoModel, AutoImageProcessor
 from PIL import Image
 
 # 1. Load model and processor
-repo_id = "Morelli001/US_UNet2DFiLM" 
+repo_id = "AImageLab-Zip/US_FiLMUNet" 
 
 processor = AutoImageProcessor.from_pretrained(repo_id, trust_remote_code=True)
 model = AutoModel.from_pretrained(repo_id, trust_remote_code=True)
@@ -115,7 +115,7 @@ This version has a U-Net depth of 4.
 from transformers import AutoModel
 
 model_4_stages = AutoModel.from_pretrained(
-    "Morelli001/US_UNet2DFiLM", 
+    "AImageLab-Zip/US_FiLMUNet", 
     subfolder="unet_4_stages",
     trust_remote_code=True
 )