ccm commited on
Commit
679a019
1 Parent(s): e994c91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('/content/modelV13_500trained_1.h5')
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]