Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ model1.add(keras.layers.UpSampling2D((1, 1)))
|
|
25 |
model1.add(keras.layers.Conv2D(3, (9, 9), activation='tanh', padding='same'))
|
26 |
|
27 |
#Loading the weights in the architecture (The file should be stored in the same directory as the code)
|
28 |
-
model1.load_weights('
|
29 |
|
30 |
def predict(mask):
|
31 |
X = numpy.round((mask/255.0))[numpy.newaxis, :, :, numpy.newaxis]
|
|
|
25 |
model1.add(keras.layers.Conv2D(3, (9, 9), activation='tanh', padding='same'))
|
26 |
|
27 |
#Loading the weights in the architecture (The file should be stored in the same directory as the code)
|
28 |
+
model1.load_weights('modelV13_500trained_1.h5')
|
29 |
|
30 |
def predict(mask):
|
31 |
X = numpy.round((mask/255.0))[numpy.newaxis, :, :, numpy.newaxis]
|