Spaces:
Runtime error
Runtime error
Commit ·
0e27ecd
1
Parent(s): 5f6bc53
minor change, removing debugger
Browse files
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
|