nahidalam commited on
Commit
b86c944
·
1 Parent(s): a8b693f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def predict_and_save(img, generator_model):
27
  return im
28
 
29
  def run(image_path):
30
- model = tf.keras.models.load_model('generator')
31
  img = tf.keras.preprocessing.image.load_img(
32
  image_path, target_size=(256, 256)
33
  )
 
27
  return im
28
 
29
  def run(image_path):
30
+ model = tf.keras.models.load_model('pretrained')
31
  img = tf.keras.preprocessing.image.load_img(
32
  image_path, target_size=(256, 256)
33
  )