LPX55 commited on
Commit
65cc34c
·
verified ·
1 Parent(s): 4f34545

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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(duration=300)
149
  def infer(
150
  image,
151
  prompt,
152
  seed=42,
153
  randomize_seed=False,
154
  true_guidance_scale=1.0,
155
- num_inference_steps=50,
156
- rewrite_prompt=True,
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=True)
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