Update app.py

app.py

@@ -345,7 +345,7 @@ def decoder(inputs, input_tensor):
 
 
 
-def unet_2( n_classes=2, height=size, width=size, channels=3, metrics = ['accuracy']):
+def unet_2( n_classes=2, height=pach_size, width=pach_size, channels=3, metrics = ['accuracy']):
     inputs = Input((height, width, channels))
     
     
@@ -582,6 +582,7 @@ def unet_enssemble(n_classes=2, height=64, width=64, channels=3, metrics = ['acc
     
     return model
 
+n_classes = 23
 model = unet_enssemble(n_classes=n_classes, height = height, width = width, channels = n_channels)