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))
|