gonzalocordova commited on
Commit
0e27ecd
·
1 Parent(s): 5f6bc53

minor change, removing debugger

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -35,7 +35,6 @@ def predict_fn(image):
35
  output = model(image)
36
 
37
  probabilities = torch.exp(output)
38
- print(probabilities)
39
  # focused (0) round probability
40
  focused_prob = round(probabilities[0][0].item(), 2)
41
  # distracted (1) round probability
 
35
  output = model(image)
36
 
37
  probabilities = torch.exp(output)
 
38
  # focused (0) round probability
39
  focused_prob = round(probabilities[0][0].item(), 2)
40
  # distracted (1) round probability