Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -229,9 +229,10 @@ if has_cuda:
|
|
229 |
clevr_model.to(device)
|
230 |
clevr_model.load_state_dict(th.load(download_model('clevr_pos'), device))
|
231 |
|
232 |
-
def compose_clevr_objects(
|
|
|
233 |
coordinates = [[float(x.split(',')[0].strip()), float(x.split(',')[1].strip())]
|
234 |
-
for x in
|
235 |
coordinates += [[-1, -1]] # add unconditional score label
|
236 |
batch_size = 1
|
237 |
guidance_scale = 10
|
|
|
229 |
clevr_model.to(device)
|
230 |
clevr_model.load_state_dict(th.load(download_model('clevr_pos'), device))
|
231 |
|
232 |
+
def compose_clevr_objects(prompt):
|
233 |
+
print(prompt)
|
234 |
coordinates = [[float(x.split(',')[0].strip()), float(x.split(',')[1].strip())]
|
235 |
+
for x in prompt.split('|')]
|
236 |
coordinates += [[-1, -1]] # add unconditional score label
|
237 |
batch_size = 1
|
238 |
guidance_scale = 10
|