Hugging Face's logo

Spaces:
umn-msi
/
fatchecker
Sleeping

App Files Community
bumble-bee commited on
Commit
89eead3
·
1 Parent(s): 0fe9d74

changed model path in predict_unet

Browse files
Files changed (1) hide show
  1. predict_unet.py +1 -1
predict_unet.py CHANGED
@@ -8,7 +8,7 @@ from unet.unet_3plus import UNet_3Plus, UNet_3Plus_DeepSup, UNet_3Plus_DeepSup_C
8
 
9
 
10
  def predict_model(input, unet_type):
11
- model_path = "weights/trained_models"
12
  h, w = 256, 256
13
  input_shape = [h, w, 1]
14
  output_channels = 1
 
8
 
9
 
10
  def predict_model(input, unet_type):
11
+ model_path = "weights"
12
  h, w = 256, 256
13
  input_shape = [h, w, 1]
14
  output_channels = 1