fixes

app.py

@@ -80,7 +80,7 @@ def swap_blip_model_cpu_gpu(device_to):
 
 def run_captioning(image):
     caption = generate_caption(blip_processor_large, blip_model_large, image).strip()
-    swap_blip_model_cpu_gpu("cpu")
+    #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):
@@ -108,13 +108,13 @@ def run_editing_quality(image, item_from, item_from_other, item_to, item_to_othe
     embs_b = torch.load(os.path.join(emb_dir, f"{item_to_selected}.pt"))
     edit_dir = (embs_b.mean(0)-embs_a.mean(0)).unsqueeze(0)
     rec_pil, edit_pil = pipe_editing(
-        original_prompt,
+        caption,
         num_inference_steps=ddim_steps,
         x_in=inverted_noise.unsqueeze(0),
         edit_dir=edit_dir,
         guidance_amount=xa_guidance,
         guidance_scale=negative_guidance_scale,
-        negative_prompt=original_prompt # use the unedited prompt for the negative prompt
+        negative_prompt=caption # use the unedited prompt for the negative prompt
     )
     return edit_pil[0]
 css = '''
@@ -157,4 +157,5 @@ Upload an image to edit it. You can try `Fast mode` with prompts, or `Quality mo
     
     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])
+    
 demo.launch()