Update main.py

main.py

@@ -50,10 +50,15 @@ async def analyze(request: Request):
             probs_chunk = softmax(output[0][0].detach().numpy())
             all_probs.append(probs_chunk)
         probs = np.mean(all_probs, axis=0)
+        
+    # Define the fixed order
+    fixed_order = ["Negative", "Neutral", "Positive"]
 
+    # Build the result using fixed order
     result = [
         {"label": config.id2label[i], "score": round(float(probs[i]), 4)}
-        for i in probs.argsort()[::-1]
+        for label in fixed_order
     ]
+
     return {"result": result}