Spaces:
Runtime error
Runtime error
Commit
·
8a0a352
1
Parent(s):
ac87f27
update
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ from core.options import AllConfigs, Options
|
|
| 26 |
from core.models import LGM
|
| 27 |
from mvdream.pipeline_mvdream import MVDreamPipeline
|
| 28 |
|
| 29 |
-
import spaces
|
| 30 |
|
| 31 |
IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
|
| 32 |
IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
|
|
@@ -100,7 +100,7 @@ pipe_image = pipe_image.to(device)
|
|
| 100 |
bg_remover = rembg.new_session()
|
| 101 |
|
| 102 |
# process function
|
| 103 |
-
|
| 104 |
def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
|
| 105 |
|
| 106 |
# seed
|
|
|
|
| 26 |
from core.models import LGM
|
| 27 |
from mvdream.pipeline_mvdream import MVDreamPipeline
|
| 28 |
|
| 29 |
+
#import spaces
|
| 30 |
|
| 31 |
IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
|
| 32 |
IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
|
|
|
|
| 100 |
bg_remover = rembg.new_session()
|
| 101 |
|
| 102 |
# process function
|
| 103 |
+
#@spaces.GPU
|
| 104 |
def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
|
| 105 |
|
| 106 |
# seed
|