ashawkey commited on
Commit
d841697
1 Parent(s): a2fa331

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,6 +13,8 @@ from safetensors.torch import load_file
13
  import rembg
14
  import gradio as gr
15
 
 
 
16
  # download checkpoints
17
  from huggingface_hub import hf_hub_download
18
  ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16_fixrot.safetensors")
@@ -27,8 +29,6 @@ from core.options import AllConfigs, Options
27
  from core.models import LGM
28
  from mvdream.pipeline_mvdream import MVDreamPipeline
29
 
30
- import spaces
31
-
32
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
33
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
34
  GRADIO_VIDEO_PATH = 'gradio_output.mp4'
 
13
  import rembg
14
  import gradio as gr
15
 
16
+ import spaces
17
+
18
  # download checkpoints
19
  from huggingface_hub import hf_hub_download
20
  ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16_fixrot.safetensors")
 
29
  from core.models import LGM
30
  from mvdream.pipeline_mvdream import MVDreamPipeline
31
 
 
 
32
  IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
33
  IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
34
  GRADIO_VIDEO_PATH = 'gradio_output.mp4'