github-actions[bot] commited on
Commit
aa4ebf1
·
1 Parent(s): 6e3355f

🚀 Deploy from GitHub Actions - 2026-02-09 14:26:36

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -338,9 +338,8 @@ async def predict_emotion(file: UploadFile = File(...)):
338
  # 3. Inference ONNX
339
  outputs = onnx_session.run(['output'], {'input': input_tensor})
340
  # scores_array = outputs[0][0]
341
- logits = outputs[0][0]
342
- probs = np.exp(logits) / np.sum(np.exp(logits))
343
- scores_array = probs
344
 
345
  # 4. Format résultats
346
  return PredictionResponse(
 
338
  # 3. Inference ONNX
339
  outputs = onnx_session.run(['output'], {'input': input_tensor})
340
  # scores_array = outputs[0][0]
341
+ raw = outputs[0][0]
342
+ scores_array = 3.0 * (1 / (1 + np.exp(-raw)))
 
343
 
344
  # 4. Format résultats
345
  return PredictionResponse(