Update app.py

app.py

@@ -48,6 +48,7 @@ def resize_img(image, max_size=1024):
     return image.resize((new_width, new_height), Image.LANCZOS)
 
 @torch.no_grad()
+@spaces.GPU(duration=85)
 def image2latent(image, latent_nudging_scalar = 1.15):
     image = pipe.image_processor.preprocess(image, height=1024, width=1024,).type(pipe.vae.dtype).to("cuda")
     latents = pipe.vae.encode(image)["latent_dist"].mean
@@ -82,7 +83,7 @@ def check_hyper_flux_lora(enable_hyper_flux):
 
 @spaces.GPU(duration=85)
 def invert_and_edit(image,
-                    source_prompt
+                    source_prompt,
                     edit_prompt, 
                     num_inversion_steps,
                     num_inference_steps,