Update main.py

main.py

@@ -333,3 +333,9 @@ def get_text_embeddings(valid_df):
             valid_text_embeddings.append(text_embeddings)
 
     return model, torch.cat(valid_text_embeddings)
+
+def get_alternative_caption(image, model, text_embeddings, captions, n=1):
+    matches = predict_caption(
+        image, model, text_embeddings, captions, n+1
+    )
+    return matches[-1]