Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -191,7 +191,7 @@ def server(input: Inputs, output: Outputs, session: Session):
|
|
| 191 |
""" Get Predictions """
|
| 192 |
# forward pass
|
| 193 |
with torch.no_grad():
|
| 194 |
-
|
| 195 |
|
| 196 |
# apply sigmoid
|
| 197 |
prob = torch.sigmoid(outputs.pred_masks.squeeze(1))
|
|
|
|
| 191 |
""" Get Predictions """
|
| 192 |
# forward pass
|
| 193 |
with torch.no_grad():
|
| 194 |
+
outputs = model(**inputs, multimask_output=False)
|
| 195 |
|
| 196 |
# apply sigmoid
|
| 197 |
prob = torch.sigmoid(outputs.pred_masks.squeeze(1))
|