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 |
)
|