akhaliq HF Staff commited on
Commit
c883349
·
verified ·
1 Parent(s): ad1eda2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -25,13 +25,13 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
25
 
26
  pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
27
  transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO",
28
- subfolder='transformer',
29
- torch_dtype=dtype,
30
- device_map='cuda'),torch_dtype=dtype).to(device)
31
 
32
  pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime",
33
- weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors",
34
- adapter_name="anime")
35
  pipe.set_adapters(["anime"], adapter_weights=[1.])
36
  pipe.fuse_lora(adapter_names=["anime"], lora_scale=1.0)
37
  pipe.unload_lora_weights()
@@ -175,7 +175,8 @@ def update_dimensions_on_upload(image):
175
  return new_width, new_height
176
 
177
 
178
- with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
179
  with gr.Column(elem_id="col-container"):
180
  gr.Markdown("# 🎨 Photo to Anime", elem_id="title")
181
  gr.Markdown(
@@ -234,4 +235,5 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
234
  outputs=[width, height]
235
  )
236
 
237
- demo.launch()
 
 
25
 
26
  pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
27
  transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO",
28
+ subfolder='transformer',
29
+ torch_dtype=dtype,
30
+ device_map='cuda'),torch_dtype=dtype).to(device)
31
 
32
  pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime",
33
+ weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors",
34
+ adapter_name="anime")
35
  pipe.set_adapters(["anime"], adapter_weights=[1.])
36
  pipe.fuse_lora(adapter_names=["anime"], lora_scale=1.0)
37
  pipe.unload_lora_weights()
 
175
  return new_width, new_height
176
 
177
 
178
+ # Removed theme=gr.themes.Soft(), css=css from gr.Blocks()
179
+ with gr.Blocks() as demo:
180
  with gr.Column(elem_id="col-container"):
181
  gr.Markdown("# 🎨 Photo to Anime", elem_id="title")
182
  gr.Markdown(
 
235
  outputs=[width, height]
236
  )
237
 
238
+ # Passed theme and css to the launch() method
239
+ demo.launch(theme=gr.themes.Soft(), css=css)