Update app.py

app.py

@@ -54,24 +54,21 @@ def invert_and_edit(image,
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     if do_inversion:
-        inverted_latents_tensor, image_latents_tensor, latent_image_ids_tensor = pipe.invert(image, num_inversion_steps=num_inversion_steps, gamma=gamma)
-        inverted_latents = gr.State(value=inverted_latents_tensor)
-        image_latents = gr.State(value=image_latents_tensor)
-        latent_image_ids = gr.State(value=latent_image_ids_tensor)
+        inverted_latents, image_latents, latent_image_ids = pipe.invert(image, num_inversion_steps=num_inversion_steps, gamma=gamma)
         do_inversion = False
     
     
     output = pipe(prompt, 
-    inverted_latents=inverted_latents.value,
-    image_latents=image_latents.value,
-    latent_image_ids=latent_image_ids.value,
+    inverted_latents=inverted_latents,
+    image_latents=image_latents,
+    latent_image_ids=latent_image_ids,
     start_timestep=start_timestep, 
     stop_timestep=stop_timestep,
     num_inference_steps=num_inversion_steps,
     eta=eta, 
     ).images[0]
         
-    return output, inverted_latents.value, image_latents.value, latent_image_ids.value, do_inversion, seed
+    return output, inverted_latents, image_latents, latent_image_ids, do_inversion, seed
 
 # UI CSS
 css = """