keysun89 commited on
Commit
9e318da
·
verified ·
1 Parent(s): 0043016

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from huggingface_hub import hf_hub_download
8
 
9
  # Load trained model weights from Hugging Face Hub
10
  weights_path = hf_hub_download(
11
- repo_id="Vizuara/unet-crack-segmentation", # ensure this matches your repo
12
- filename="unet_weights_v2.pth" # make sure this file exists in repo
13
  )
14
 
15
  # Initialize and load model
 
8
 
9
  # Load trained model weights from Hugging Face Hub
10
  weights_path = hf_hub_download(
11
+ repo_id="keysun89/IMG_cv", # ensure this matches your repo
12
+ filename="residual_unet_model_x.pt" # make sure this file exists in repo
13
  )
14
 
15
  # Initialize and load model