flamehaze1115 commited on
Commit
e5c095e
·
verified ·
1 Parent(s): d5384e5

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -1
gradio_app.py CHANGED
@@ -57,7 +57,7 @@ if not hasattr(Image, 'Resampling'):
57
 
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
 
 
57
 
58
 
59
  def sam_init():
60
+ model = SamModel.from_pretrained("facebook/sam-vit-huge")
61
  processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
62
  return model, processor
63