Update app.py

app.py

@@ -74,13 +74,8 @@ def generate_inversion(prompt, image, num_ddim_steps=50):
         )
     return x_inv[0]
 
-def swap_blip_model_cpu_gpu(device_to):
-    if torch.cuda.is_available():
-        blip_model_large.to(device_to)
-
 def run_captioning(image):
     caption = generate_caption(blip_processor_large, blip_model_large, image).strip()
-    #swap_blip_model_cpu_gpu("cpu")
     return caption
     
 def run_editing(image, original_prompt, edit_prompt, ddim_steps=50, xa_guidance=0.1, negative_guidance_scale=5.0):
@@ -149,14 +144,17 @@ Upload an image to edit it. You can try `Fast mode` with prompts, or `Quality mo
                             item_to_other = gr.Textbox(visible=False, label="Type what to replace what you identified for")
                             item_to.change(lambda choice: gr.Dropdown.update(visible=choice=="Other"), item_to, item_to_other)
                     btn_edit_quality = gr.Button("Edit Image")    
-                        
+            with gr.Accordion(label="Advanced settings", open=False):
+                steps = gr.Slider(minimum=2, maximum=50, step=1, value=50)
+                xa_guidance =gr.Slider(minimum=0.0, maximum=10.0, step=0.05, value=0.1)
+                negative_scale = gr.Slider(minimum=0.0, maximum=20.0, step=0.1, value=5.0)
         with gr.Column():
             image_output = gr.Image(label="Image with edits",type="pil",shape=(512, 512))
             
     btn_caption.click(fn=run_captioning, inputs=image, outputs=original_prompt)
     
-    btn_edit_fast.click(fn=run_editing, inputs=[image, original_prompt, edit_prompt], outputs=[image_output])
-    btn_edit_quality.click(fn=run_editing_quality, inputs=[image, item_from, item_from_other, item_to, item_to_other], outputs=[image_output])
+    btn_edit_fast.click(fn=run_editing, inputs=[image, original_prompt, edit_prompt, steps, xa_guidance, negative_scale], outputs=[image_output])
+    btn_edit_quality.click(fn=run_editing_quality, inputs=[image, item_from, item_from_other, item_to, item_to_other, steps, xa_guidance, negative_scale], outputs=[image_output])
     gr.Examples(
         examples=[
             [os.path.join(os.path.dirname(__file__), "assets/test_images/cats/cat_1.png"), "cat", "", "dog", ""],