Update app.py

app.py

@@ -68,7 +68,7 @@ def invert_and_edit(image,
     eta=eta, 
     ).images[0]
         
-    return output, inverted_latents, image_latents, latent_image_ids, do_inversion, seed
+    return output, inverted_latents.cpu(), image_latents.cpu(), latent_image_ids.cpu(), do_inversion, seed
 
 # UI CSS
 css = """