update examples
Browse files
app.py
CHANGED
|
@@ -534,7 +534,13 @@ def run_one(paths: Paths, prompt: str, steps: int = DEFAULT_STEPS, category: str
|
|
| 534 |
|
| 535 |
style_img = Image.open(paths.style_path).convert("RGB")
|
| 536 |
|
| 537 |
-
prompt = extractor_category + "with " + prompt
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 538 |
print("==== prompt? ", prompt, flush=True)
|
| 539 |
|
| 540 |
with torch.inference_mode():
|
|
|
|
| 534 |
|
| 535 |
style_img = Image.open(paths.style_path).convert("RGB")
|
| 536 |
|
| 537 |
+
prompt = extractor_category + " with " + prompt
|
| 538 |
+
|
| 539 |
+
if prompt != "":
|
| 540 |
+
prompt = extractor_category + " with " + prompt
|
| 541 |
+
else:
|
| 542 |
+
prompt = extractor_category
|
| 543 |
+
|
| 544 |
print("==== prompt? ", prompt, flush=True)
|
| 545 |
|
| 546 |
with torch.inference_mode():
|