rynmurdock commited on
Commit
2927b74
1 Parent(s): 5fc21dd

basically delirious lol

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -126,7 +126,7 @@ pipe.to(device=DEVICE)
126
  #pipe.unet = torch.compile(pipe.unet)
127
  #pipe.vae = torch.compile(pipe.vae)
128
 
129
- @spaces.GPU(duration=5)
130
  def generate_gpu(in_im_embs):
131
  print('start gen')
132
  in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
@@ -222,9 +222,7 @@ def pluck_img(user_id, user_emb):
222
 
223
 
224
  def background_next_image():
225
- global prevs_df
226
- global is_started
227
- while True:
228
  # only let it get N (maybe 3) ahead of the user
229
  #not_rated_rows = prevs_df[[i[1]['user:rating'] == {' ': ' '} for i in prevs_df.iterrows()]]
230
  rated_rows = prevs_df[[i[1]['user:rating'] != {' ': ' '} for i in prevs_df.iterrows()]]
 
126
  #pipe.unet = torch.compile(pipe.unet)
127
  #pipe.vae = torch.compile(pipe.vae)
128
 
129
+
130
  def generate_gpu(in_im_embs):
131
  print('start gen')
132
  in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
 
222
 
223
 
224
  def background_next_image():
225
+ global prevs_df
 
 
226
  # only let it get N (maybe 3) ahead of the user
227
  #not_rated_rows = prevs_df[[i[1]['user:rating'] == {' ': ' '} for i in prevs_df.iterrows()]]
228
  rated_rows = prevs_df[[i[1]['user:rating'] != {' ': ' '} for i in prevs_df.iterrows()]]