Spaces:
Build error
Build error
Trang Dang
commited on
Commit
•
199a681
1
Parent(s):
0b3ff07
no cache
Browse files
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)
|