Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -145,15 +145,15 @@ pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype
|
|
| 145 |
MAX_SEED = np.iinfo(np.int32).max
|
| 146 |
|
| 147 |
# --- Main Inference Function (with hardcoded negative prompt) ---
|
| 148 |
-
@spaces.GPU(
|
| 149 |
def infer(
|
| 150 |
image,
|
| 151 |
prompt,
|
| 152 |
seed=42,
|
| 153 |
randomize_seed=False,
|
| 154 |
true_guidance_scale=1.0,
|
| 155 |
-
num_inference_steps=
|
| 156 |
-
rewrite_prompt=
|
| 157 |
num_images_per_prompt=1,
|
| 158 |
progress=gr.Progress(track_tqdm=True),
|
| 159 |
):
|
|
@@ -258,7 +258,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 258 |
value=1,
|
| 259 |
)
|
| 260 |
|
| 261 |
-
rewrite_prompt = gr.Checkbox(label="Rewrite prompt", value=
|
| 262 |
|
| 263 |
# gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False)
|
| 264 |
|
|
|
|
| 145 |
MAX_SEED = np.iinfo(np.int32).max
|
| 146 |
|
| 147 |
# --- Main Inference Function (with hardcoded negative prompt) ---
|
| 148 |
+
@spaces.GPU()
|
| 149 |
def infer(
|
| 150 |
image,
|
| 151 |
prompt,
|
| 152 |
seed=42,
|
| 153 |
randomize_seed=False,
|
| 154 |
true_guidance_scale=1.0,
|
| 155 |
+
num_inference_steps=20,
|
| 156 |
+
rewrite_prompt=False,
|
| 157 |
num_images_per_prompt=1,
|
| 158 |
progress=gr.Progress(track_tqdm=True),
|
| 159 |
):
|
|
|
|
| 258 |
value=1,
|
| 259 |
)
|
| 260 |
|
| 261 |
+
rewrite_prompt = gr.Checkbox(label="Rewrite prompt", value=False, visible=False)
|
| 262 |
|
| 263 |
# gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False)
|
| 264 |
|