run.py update device

run.py

@@ -9,6 +9,9 @@ import PIL
 def pred(src):
     # os.environ['HUGGINGFACE_HUB_HOME'] = './.cache'
     # Load the model configuration
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    model.to(device)
+    
     cache_dir = "/code/cache"
     model_config = SamConfig.from_pretrained("facebook/sam-vit-base", 
                                              cache_dir=cache_dir)