hysts HF staff commited on
Commit
c37294a
1 Parent(s): 5719c29

Remove unused import

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -8,10 +8,10 @@ import spaces
8
  import torch
9
  import torchvision
10
  from diffusers import AutoencoderKL, DDIMScheduler
11
- from einops import rearrange, repeat
12
  from omegaconf import OmegaConf
13
  from PIL import Image
14
- from torchvision import transforms, utils
15
  from transformers import CLIPTextModel, CLIPTokenizer
16
 
17
  from modules.unet import UNet3DConditionFlowModel
@@ -25,9 +25,8 @@ from utils.utils import (
25
  interpolate_trajectory,
26
  load_model,
27
  load_weights,
28
- save_videos_grid,
29
  )
30
- from utils.visualizer import Visualizer, vis_flow_to_video
31
 
32
  #### Description ####
33
  title = r"""<h1 align="center">CustomNet: Object Customization with Variable-Viewpoints in Text-to-Image Diffusion Models</h1>"""
 
8
  import torch
9
  import torchvision
10
  from diffusers import AutoencoderKL, DDIMScheduler
11
+ from einops import rearrange
12
  from omegaconf import OmegaConf
13
  from PIL import Image
14
+ from torchvision import transforms
15
  from transformers import CLIPTextModel, CLIPTokenizer
16
 
17
  from modules.unet import UNet3DConditionFlowModel
 
25
  interpolate_trajectory,
26
  load_model,
27
  load_weights,
 
28
  )
29
+ from utils.visualizer import vis_flow_to_video
30
 
31
  #### Description ####
32
  title = r"""<h1 align="center">CustomNet: Object Customization with Variable-Viewpoints in Text-to-Image Diffusion Models</h1>"""