Spaces:
Running
on
Zero
Running
on
Zero
update to v1.5
Browse files- gradio_app.py +1 -4
gradio_app.py
CHANGED
@@ -67,7 +67,6 @@ If you have any questions, feel free to open a discussion or contact us at <b>we
|
|
67 |
model = None
|
68 |
cached_dir = None
|
69 |
|
70 |
-
generator = None
|
71 |
|
72 |
def check_input_image(input_image):
|
73 |
if input_image is None:
|
@@ -135,7 +134,7 @@ class RMBG(object):
|
|
135 |
image.paste(padded_image, paste_position)
|
136 |
return image
|
137 |
|
138 |
-
|
139 |
def image2mesh(image: Any,
|
140 |
more: bool = False,
|
141 |
scheluder_name: str ="DDIMScheduler",
|
@@ -203,8 +202,6 @@ if __name__=="__main__":
|
|
203 |
"Original Image": "Original Image",
|
204 |
})
|
205 |
|
206 |
-
generator = torch.Generator(device)
|
207 |
-
|
208 |
# for 3D latent set diffusion
|
209 |
if args.model_path == "":
|
210 |
ckpt_path = hf_hub_download(repo_id="craftsman3d/craftsman-v1-5", filename="model.ckpt", repo_type="model")
|
|
|
67 |
model = None
|
68 |
cached_dir = None
|
69 |
|
|
|
70 |
|
71 |
def check_input_image(input_image):
|
72 |
if input_image is None:
|
|
|
134 |
image.paste(padded_image, paste_position)
|
135 |
return image
|
136 |
|
137 |
+
@spaces.GPU
|
138 |
def image2mesh(image: Any,
|
139 |
more: bool = False,
|
140 |
scheluder_name: str ="DDIMScheduler",
|
|
|
202 |
"Original Image": "Original Image",
|
203 |
})
|
204 |
|
|
|
|
|
205 |
# for 3D latent set diffusion
|
206 |
if args.model_path == "":
|
207 |
ckpt_path = hf_hub_download(repo_id="craftsman3d/craftsman-v1-5", filename="model.ckpt", repo_type="model")
|