Update app.py

app.py

@@ -23,7 +23,6 @@ pipeline.enable_model_cpu_offload()
 pipeline.enable_vae_slicing()
 
 def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
-    
     # DDIM inversion
     src_style = f"{ref_style}"
     src_prompt = f"{ref_prompt}, {src_style}."
@@ -31,7 +30,11 @@ def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
 
     num_inference_steps = 50
     x0 = np.array(load_image(image_path).resize((1024, 1024)))
-    zts = inversion.ddim_inversion(pipeline, x0, src_prompt, num_inference_steps, 2)
+
+    try:
+        zts = inversion.ddim_inversion(pipeline, x0, src_prompt, num_inference_steps, 2)
+    except:
+        zts = inversion.ddim_inversion(pipeline, x0, src_prompt, num_inference_steps, 2)
     #mediapy.show_image(x0, title="innput reference image", height=256)
 
     # run StyleAligned
@@ -88,7 +91,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
             ref_style = gr.Textbox(label="Reference style", value="medieval painting")
-            ref_prompt = gr.Textbox(label="Referéce prompt", value="Man laying on bed")
+            ref_prompt = gr.Textbox(label="Reference prompt", value="Man laying on bed")
             
         with gr.Column():