Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -210,11 +210,10 @@ vectorization.adapt(data_txt)
|
|
210 |
|
211 |
def generate_caption(image):
|
212 |
|
213 |
-
|
214 |
-
|
215 |
-
|
216 |
-
|
217 |
-
|
218 |
img = tf.expand_dims(img, 0)
|
219 |
img = loaded_model.cnn_model(img)
|
220 |
encoded_img = loaded_model.encoder(img, training=False)
|
|
|
210 |
|
211 |
def generate_caption(image):
|
212 |
|
213 |
+
|
214 |
+
img = tf.constant(image)
|
215 |
+
img = tf.image.resize(img, IMAGE_SIZE)
|
216 |
+
img = tf.image.convert_image_dtype(img, tf.float32)
|
|
|
217 |
img = tf.expand_dims(img, 0)
|
218 |
img = loaded_model.cnn_model(img)
|
219 |
encoded_img = loaded_model.encoder(img, training=False)
|