Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def apply_gaussian_blur(image):
|
|
20 |
|
21 |
# Perform semantic segmentation using the model
|
22 |
with torch.no_grad():
|
23 |
-
outputs =
|
24 |
logits = outputs.logits
|
25 |
|
26 |
# Get the predicted class for each pixel
|
|
|
20 |
|
21 |
# Perform semantic segmentation using the model
|
22 |
with torch.no_grad():
|
23 |
+
outputs = segmentation_model(**inputs)
|
24 |
logits = outputs.logits
|
25 |
|
26 |
# Get the predicted class for each pixel
|