Update app.py

app.py

@@ -40,15 +40,25 @@ label_map = {
     "LABEL_26": "Composite3"
 }
 
-# Prediction function
 def classify_numbers(input1, input2, input3):
-    # Combine inputs into a single string (or modify to match your training format)
-    text = f"{input1}, {input2}, {input3}"
-    result = classifier(text)
+    inputs = [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
+
+        # 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))
+        }
+
+        predictions[f"Input{number_index}"] = filtered
+
+    return predictions
 
-    # Convert raw LABEL_X output into readable format
-    output = {label_map[item["label"]]: round(item["score"], 3) for item in result[0]}
-    return output
 
 # Gradio UI
 iface = gr.Interface(