fffiloni commited on
Commit
51db498
·
verified ·
1 Parent(s): 6197571

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +3 -1
gradio_app.py CHANGED
@@ -1,10 +1,12 @@
1
  import os
2
  import gradio as gr
3
  import torch
 
4
 
5
  # import argparse
6
 
7
- checkpoint_dir = "checkpoints/svd_reverse_motion_with_attnflip/unet"
 
8
 
9
  from diffusers.utils import load_image, export_to_video
10
  from diffusers import UNetSpatioTemporalConditionModel
 
1
  import os
2
  import gradio as gr
3
  import torch
4
+ from huggingface_hub import snapshot_download
5
 
6
  # import argparse
7
 
8
+ snapshot_download(repo_id="fffiloni/svd_keyframe_interpolation", local_dir="checkpoints")
9
+ checkpoint_dir = "checkpoints/svd_reverse_motion_with_attnflip"
10
 
11
  from diffusers.utils import load_image, export_to_video
12
  from diffusers import UNetSpatioTemporalConditionModel