Update app.py

app.py

@@ -110,9 +110,10 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
         num_channels_latents = int(num_channels_latents / 2)
         image = Image.open(image.name).convert('RGB')
         image.resize((size,size), Image.LANCZOS)
-        image = pipe.video_processor.preprocess(image, height=size, width=size).to(
+        with torch.no_grad():
+            image = pipe.video_processor.preprocess(image, height=size, width=size).to(
                 device, dtype=prompt_embeds.dtype
-        )
+            )
         num_latent_frames = (frames - 1) // pipe.vae_scale_factor_temporal + 1
         latents = pipe.prepare_latents(
             batch_size=1, num_channels_latents=pipe.transformer.config.in_channels, height=size, width=size, num_frames=frames,