Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -397,7 +397,7 @@ def infer(
|
|
397 |
prompt_with_template= prompt
|
398 |
|
399 |
if rewrite_prompt:
|
400 |
-
final_prompt = polish_prompt(prompt_with_template,
|
401 |
else:
|
402 |
final_prompt = prompt_with_template
|
403 |
|
|
|
397 |
prompt_with_template= prompt
|
398 |
|
399 |
if rewrite_prompt:
|
400 |
+
final_prompt = polish_prompt(prompt_with_template, image)
|
401 |
else:
|
402 |
final_prompt = prompt_with_template
|
403 |
|