rynmurdock commited on
Commit
c847977
·
1 Parent(s): 8f8d78c
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def write_video(file_name, images, fps=17):
60
 
61
  stream = container.add_stream("h264", rate=fps)
62
  # stream.options = {'preset': 'faster'}
63
- stream.thread_count = 0
64
  stream.width = 512
65
  stream.height = 512
66
  stream.pix_fmt = "yuv420p"
@@ -209,6 +209,7 @@ def pluck_img(user_id, user_emb):
209
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
210
  while len(not_rated_rows) == 0:
211
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
 
212
  # TODO optimize this lol
213
  best_sim = -100000
214
  for i in not_rated_rows.iterrows():
 
60
 
61
  stream = container.add_stream("h264", rate=fps)
62
  # stream.options = {'preset': 'faster'}
63
+ stream.thread_count = 1
64
  stream.width = 512
65
  stream.height = 512
66
  stream.pix_fmt = "yuv420p"
 
209
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
210
  while len(not_rated_rows) == 0:
211
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
212
+ time.sleep(1)
213
  # TODO optimize this lol
214
  best_sim = -100000
215
  for i in not_rated_rows.iterrows():