Update gradio_app.py

gradio_app.py

@@ -57,8 +57,8 @@ if not hasattr(Image, 'Resampling'):
 
 
 def sam_init():
-    model = SamModel.from_pretrained("facebook/sam-vit-huge")
-    processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
+    model = SamModel.from_pretrained("facebook/sam-vit-huge").to(device=f"cuda:{_GPU_ID}")
+    processor = SamProcessor.from_pretrained("facebook/sam-vit-huge").to(device=f"cuda:{_GPU_ID}")
     return model, processor
 
 def sam_segment(sam_model, sam_processor, input_image, *bbox_coords):