xrg commited on
Commit
96af654
·
1 Parent(s): e4a8c3a
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -23,6 +23,8 @@ from mvdream.pipeline_mvdream import MVDreamPipeline
23
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
24
  from huggingface_hub import hf_hub_download
25
 
 
 
26
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
27
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
28
  GRADIO_VIDEO_PATH = 'gradio_output.mp4'
 
23
  from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
24
  from huggingface_hub import hf_hub_download
25
 
26
+ import spaces
27
+
28
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
29
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
30
  GRADIO_VIDEO_PATH = 'gradio_output.mp4'