Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ ENCODER_WEIGHTS = 'imagenet'
|
|
25 |
# Load and prepare the model
|
26 |
@st.cache_resource
|
27 |
def load_model():
|
28 |
-
model = torch.load(
|
29 |
model.eval().float()
|
30 |
return model
|
31 |
|
|
|
25 |
# Load and prepare the model
|
26 |
@st.cache_resource
|
27 |
def load_model():
|
28 |
+
model = torch.load('deeplabv3+ v15.pth', map_location=DEVICE)
|
29 |
model.eval().float()
|
30 |
return model
|
31 |
|