Update app.py

app.py

@@ -30,12 +30,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 text_encoder = CLIPTextModel.from_pretrained(model_repo_id, subfolder="text_encoder")
 vae = AutoencoderKL.from_pretrained(model_repo_id, subfolder="vae")
 unet = UNet2DConditionModel.from_config(model_repo_id, subfolder="unet")
-try:
-    medsegfactory_ckpt = hf_hub_download(repo_id=medsegfactory_id, filename=filename)
-    unet.load_state_dict(torch.load(medsegfactory_ckpt, map_location='cpu'))
-except Exception as e:
-    print("[Error] Failed to load checkpoint:", e)
-    pipeline = None
+
+medsegfactory_ckpt = hf_hub_download(repo_id=medsegfactory_id, filename=filename)
+unet.load_state_dict(torch.load(medsegfactory_ckpt, map_location='cpu'))
+
 vae.requires_grad_(False)
 text_encoder.requires_grad_(False)