Spaces:

flax-community
/

image-captioning

Runtime error

ydshieh commited on Jul 18, 2021

Commit

3ca5ee2

•

1 Parent(s): 13b9f3b

fix caption

Files changed (1) hide show

model.py CHANGED Viewed

@@ -53,7 +53,7 @@ def predict(image):
     token_ids = np.array(generation.sequences)[0]
     caption = tokenizer.decode(token_ids)
     caption = caption.replace('<s>', '').replace('</s>', '').replace('<pad>', '')
-    caption = caption.replace("à l'arrière-plan", '').replace("Une photo en noir et blanc d", '').replace("Une photo noire et blanche d'", '').replace("en arrière-plan", '')
     while '  ' in caption:
         caption = caption.replace('  ', ' ')
     caption = caption.strip()

     token_ids = np.array(generation.sequences)[0]
     caption = tokenizer.decode(token_ids)
     caption = caption.replace('<s>', '').replace('</s>', '').replace('<pad>', '')
+    caption = caption.replace("à l'arrière-plan", '').replace("Une photo en noir et blanc d'", '').replace("Une photo noire et blanche d'", '').replace("en arrière-plan", '')
     while '  ' in caption:
         caption = caption.replace('  ', ' ')
     caption = caption.strip()