Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
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
|