wyysf commited on
Commit
c52f8a7
1 Parent(s): 8133633

update to v1.5

Browse files
Files changed (1) hide show
  1. 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
- # @spaces.GPU
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")