windy2612 commited on
Commit
7bed5cc
·
verified ·
1 Parent(s): 8a341bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -210,11 +210,10 @@ vectorization.adapt(data_txt)
210
 
211
  def generate_caption(image):
212
 
213
- if isinstance(image, np.ndarray):
214
- img = tf.constant(image)
215
- img = tf.image.resize(img, IMAGE_SIZE)
216
- img = tf.image.convert_image_dtype(img, tf.float32)
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)