Update app.py

app.py

@@ -14,8 +14,8 @@ def tune_video_predict(
     num_inference_steps: int,
     guidance_scale: float,
 ):
-    unet = UNet3DConditionModel.from_pretrained(pipe_id, subfolder='unet', torch_dtype=torch.float16).to('cuda')
-    pipe = TuneAVideoPipeline.from_pretrained('CompVis/stable-diffusion-v1-4', unet=unet, torch_dtype=torch.float16).to("cuda")
+    unet = UNet3DConditionModel.from_pretrained("Tune-A-Video-library/a-man-is-surfing", subfolder='unet', torch_dtype=torch.float16).to('cuda')
+    pipe = TuneAVideoPipeline.from_pretrained(pipe_id, unet=unet, torch_dtype=torch.float16).to("cuda")
     video = pipe(prompt, video_length=video_length, height=height, width=width, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).videos
     output_path = save_videos_grid(video, save_path='output', path=f"{prompt}.gif")
     return output_path