Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -503,10 +503,23 @@ def generate(prompt, vp_pos, vp_neg, slider, example_prompt, gallery, num_infere
|
|
| 503 |
if re.search(rf"\b{filter}\b", vp_pos) or re.search(rf"\b{filter}\b", vp_neg):
|
| 504 |
raise gr.Error("Please try again with a different visual preference")
|
| 505 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 506 |
if randomize_seed:
|
| 507 |
seed = random.randint(0, MAX_SEED)
|
| 508 |
|
| 509 |
generator = torch.Generator().manual_seed(seed)
|
|
|
|
|
|
|
|
|
|
| 510 |
|
| 511 |
image = pipe(prompt=prompt,
|
| 512 |
num_inference_steps=num_inference_steps,
|
|
|
|
| 503 |
if re.search(rf"\b{filter}\b", vp_pos) or re.search(rf"\b{filter}\b", vp_neg):
|
| 504 |
raise gr.Error("Please try again with a different visual preference")
|
| 505 |
|
| 506 |
+
if vp_pos != "":
|
| 507 |
+
vp_pos = vp_pos.split(", ")
|
| 508 |
+
random.shuffle(vp_pos)
|
| 509 |
+
vp_pos = ", ".join(vp_pos)
|
| 510 |
+
|
| 511 |
+
if vp_neg != "":
|
| 512 |
+
vp_neg = vp_neg.split(", ")
|
| 513 |
+
random.shuffle(vp_neg)
|
| 514 |
+
vp_neg = ", ".join(vp_neg)
|
| 515 |
+
|
| 516 |
if randomize_seed:
|
| 517 |
seed = random.randint(0, MAX_SEED)
|
| 518 |
|
| 519 |
generator = torch.Generator().manual_seed(seed)
|
| 520 |
+
|
| 521 |
+
if prompt is None:
|
| 522 |
+
promtpt = ""
|
| 523 |
|
| 524 |
image = pipe(prompt=prompt,
|
| 525 |
num_inference_steps=num_inference_steps,
|