Update model

app.py

@@ -11,9 +11,9 @@ import lwbna_unet as u
 MAX_SELECTIONS = 8
 input_shape = (MAX_SELECTIONS,256,256,2)
 def load_model():
-    unet = u.LWBNAUnet(1,64,4,4)
+    unet = u.LWBNAUnet(1,128,8,4)
     unet.build(input_shape)
-    unet.load_weights('robust-energy-model-best.hdf5')
+    unet.load_weights('royal-snowflake-2400.hdf5')
     return unet
 
 model = load_model()