Spaces:
Runtime error
Runtime error
Commit
·
8386247
1
Parent(s):
f9c1121
basically delirious lol
Browse files
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 |
-
|
| 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 |
-
|
| 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()]]
|