Hugging Face's logo

Spaces:
peshk1n
/
image-captioning
Sleeping

App Files Community
peshk1n commited on
Commit
dcbbe48
·
verified ·
1 Parent(s): 6c80bc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -601,7 +601,6 @@ def load_and_preprocess_image(img):
601
 
602
  def generate_caption_coca(image):
603
  img_processed = load_and_preprocess_image(image)
604
- img_processed = np.expand_dims(img_processed, axis=0)
605
 
606
  _, cap_features = coca_model.encoder.predict(img_processed, verbose=0)
607
  cap_features = cap_features.astype(np.float32)
 
601
 
602
  def generate_caption_coca(image):
603
  img_processed = load_and_preprocess_image(image)
 
604
 
605
  _, cap_features = coca_model.encoder.predict(img_processed, verbose=0)
606
  cap_features = cap_features.astype(np.float32)