Update app.py
Browse files
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('
|
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 |
)
|