Spaces:
Runtime error
Runtime error
Update model
Browse files
app.py
CHANGED
@@ -11,9 +11,9 @@ import lwbna_unet as u
|
|
11 |
MAX_SELECTIONS = 8
|
12 |
input_shape = (MAX_SELECTIONS,256,256,2)
|
13 |
def load_model():
|
14 |
-
unet = u.LWBNAUnet(1,
|
15 |
unet.build(input_shape)
|
16 |
-
unet.load_weights('
|
17 |
return unet
|
18 |
|
19 |
model = load_model()
|
|
|
11 |
MAX_SELECTIONS = 8
|
12 |
input_shape = (MAX_SELECTIONS,256,256,2)
|
13 |
def load_model():
|
14 |
+
unet = u.LWBNAUnet(1,128,8,4)
|
15 |
unet.build(input_shape)
|
16 |
+
unet.load_weights('royal-snowflake-2400.hdf5')
|
17 |
return unet
|
18 |
|
19 |
model = load_model()
|