Update main.py

main.py

@@ -56,7 +56,7 @@ async def analyze(request: Request):
 
     # Build the result using fixed order
     result = [
-        {"label": config.id2label[i], "score": round(float(probs[i]), 4)}
+        {"label": label, "score": round(float(probs[config.label2id[label]]), 4)}
         for label in fixed_order
     ]