Cannot run the simple_retrieval

[findalexli]

Hello there,

I was trying to run the retrieval with Cohere on the simple retireval script, but it just constantly look though multiple for loop of 2048 without ending