Linoy Tsaban commited on
Commit
99731b2
1 Parent(s): 6443d39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -2,8 +2,10 @@ import gradio as gr
2
  import torch
3
  from diffusers import StableDiffusionPipeline, DDIMScheduler
4
  from utils import video_to_frames, add_dict_to_yaml_file, save_video
5
- from diffusers.utils import export_to_video
6
-
 
 
7
  # load sd model
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
  # model_id = "stabilityai/stable-diffusion-2-1-base"
 
2
  import torch
3
  from diffusers import StableDiffusionPipeline, DDIMScheduler
4
  from utils import video_to_frames, add_dict_to_yaml_file, save_video
5
+ # from diffusers.utils import export_to_video
6
+ from tokenflow_pnp import TokenFlow
7
+ from preprocess_utils import *
8
+ from tokenflow_utils import *
9
  # load sd model
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  # model_id = "stabilityai/stable-diffusion-2-1-base"