ciro-c commited on
Commit
606eac6
·
1 Parent(s): 9594094

Fixing result error

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,8 @@ def predict(text):
21
 
22
  outputs = model(**inputs)
23
  predictions = outputs.logits
24
- result = np.exp(predictions.item())
 
25
  result = max(result, 0)
26
  result = min(result, 5)
27
  return mapping[round(result)]
 
21
 
22
  outputs = model(**inputs)
23
  predictions = outputs.logits
24
+ predictionItem = predictions.item()
25
+ result = torch.sum(predictionItem)
26
  result = max(result, 0)
27
  result = min(result, 5)
28
  return mapping[round(result)]