Untitled

def find_image(row):
    to_predict1 = pd.concat([test_images_vector, 
                            pd.concat([row] * len(test_images_vector), 
                                      ignore_index=True)], axis=1)
    return test_images. iloc[np.argmax(Ridge_best.predict(to_predict1))][0]


test_text_vector.apply(find_image, axis=1)