Update app.py
Browse files
app.py
CHANGED
@@ -272,11 +272,13 @@ class ImageCaptioningModel(tf.keras.Model):
|
|
272 |
def load_image_from_path(img_path):
|
273 |
img = tf.io.read_file(img_path)
|
274 |
img = tf.io.decode_jpeg(img, channels=3)
|
|
|
275 |
img = tf.keras.layers.Resizing(299, 299)(img)
|
276 |
img = tf.keras.applications.inception_v3.preprocess_input(img)
|
277 |
return img
|
278 |
|
279 |
|
|
|
280 |
def generate_caption(img, caption_model, add_noise=False):
|
281 |
if isinstance(img, str):
|
282 |
img = load_image_from_path(img)
|
|
|
272 |
def load_image_from_path(img_path):
|
273 |
img = tf.io.read_file(img_path)
|
274 |
img = tf.io.decode_jpeg(img, channels=3)
|
275 |
+
img = tf.cast(img, tf.float32) # Convert to float32
|
276 |
img = tf.keras.layers.Resizing(299, 299)(img)
|
277 |
img = tf.keras.applications.inception_v3.preprocess_input(img)
|
278 |
return img
|
279 |
|
280 |
|
281 |
+
|
282 |
def generate_caption(img, caption_model, add_noise=False):
|
283 |
if isinstance(img, str):
|
284 |
img = load_image_from_path(img)
|