Update app.py

app.py

@@ -45,21 +45,28 @@ def classify_numbers(input1, input2, input3):
     predictions = {}
 
     for i, value in enumerate(inputs):
-        result = classifier(value)[0]  # get top_k=None full result
-        number_index = i + 1  # 1-based index: 1, 2, 3
+        result = classifier(value)[0]  # full list of label scores
+        number_index = str(i + 1)
 
-        # Filter and rename labels specific to this input
-        filtered = {
-            label_map[item["label"]]: round(item["score"], 3)
-            for item in result
-            if label_map[item["label"]].endswith(str(number_index))
-        }
+        input_result = {}
 
-        predictions[f"Input{number_index}"] = filtered
+        for item in result:
+            label = label_map[item["label"]]
+            if label.endswith(number_index):
+                input_result[label] = round(item["score"], 3)
+
+        # ✅ Sort the results by score (highest to lowest)
+        input_result = dict(
+            sorted(input_result.items(), key=lambda x: x[1], reverse=True)
+        )
+
+        predictions[f"Input{number_index}"] = input_result
 
     return predictions
 
 
+
+
 # Gradio UI
 iface = gr.Interface(
     fn=classify_numbers,