Update main.py

main.py

@@ -333,9 +333,3 @@ 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]