.

app.py

@@ -54,7 +54,7 @@ from transformers import CLIPVisionModelWithProjection
 import uuid
 import av
 
-def write_video_av(file_name, images, fps=17):
+def write_video(file_name, images, fps=17):
     print('Saving')
     container = av.open(file_name, mode="w")
 
@@ -78,7 +78,7 @@ def write_video_av(file_name, images, fps=17):
     container.close()
     print('Saved')
 
-def write_video(file_name, images, fps=15):
+def imio_write_video(file_name, images, fps=15):
     writer = imageio.get_writer(file_name, fps=fps)
 
     for im in images:
@@ -209,8 +209,6 @@ def pluck_img(user_id, user_emb):
     not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
     while len(not_rated_rows) == 0:
         not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
-        time.sleep(.1)
-        print(f'user {user_id} has rated all rows')
     # TODO optimize this lol
     best_sim = -100000
     for i in not_rated_rows.iterrows():