Trang Dang commited on
Commit
199a681
1 Parent(s): 0b3ff07
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ def server(input: Inputs, output: Outputs, session: Session):
48
  img = {"src": src, "width": "500px"}
49
 
50
  # Load the model configuration
51
- model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
52
- processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
53
 
54
  # Create an instance of the model architecture with the loaded configuration
55
  my_sam_model = SamModel(config=model_config)
 
48
  img = {"src": src, "width": "500px"}
49
 
50
  # Load the model configuration
51
+ model_config = SamConfig.from_pretrained("facebook/sam-vit-base", use_cache=False)
52
+ processor = SamProcessor.from_pretrained("facebook/sam-vit-base", use_cache=False)
53
 
54
  # Create an instance of the model architecture with the loaded configuration
55
  my_sam_model = SamModel(config=model_config)