aagoluoglu commited on
Commit
8f831ae
·
verified ·
1 Parent(s): db5d62c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- outputs = model(**inputs, multimask_output=False)
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))