Hugging Face's logo

Spaces:
flamehaze1115
/
Wonder3D-demo
Runtime error

App Files Community
24
flamehaze1115 commited on
Commit
2ae688f
·
verified ·
1 Parent(s): 7f8cd5d

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -58,7 +58,7 @@ if not hasattr(Image, 'Resampling'):
58
 
59
  def sam_init():
60
  model = SamModel.from_pretrained("facebook/sam-vit-huge").to(device=f"cuda:{_GPU_ID}")
61
- processor = SamProcessor.from_pretrained("facebook/sam-vit-huge").to(device=f"cuda:{_GPU_ID}")
62
  return model, processor
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):
 
58
 
59
  def sam_init():
60
  model = SamModel.from_pretrained("facebook/sam-vit-huge").to(device=f"cuda:{_GPU_ID}")
61
+ processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
62
  return model, processor
63
 
64
  def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):