Update app.py

app.py

@@ -36,14 +36,17 @@ def classify_image(inp):
     prediction = model.predict(inp)
     predicted_class_indices = np.argmax(prediction, axis=1)
     result = {}
-    for i in range(len(predicted_class_indices)):
-        if predicted_class_indices[i] < NUM_CLASSES:
-            try:
-                label = labels[predicted_class_indices[i]]
-                result[label] = float(predicted_class_indices[i])
-            except KeyError:
-                print(f"KeyError: Label not found for index {predicted_class_indices[i]}")
-    return f"{labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}" 
+    # for i in range(len(predicted_class_indices)):
+    #     if predicted_class_indices[i] < NUM_CLASSES:
+    #         try:
+    #             label = labels[predicted_class_indices[i]]
+    #             result[label] = float(predicted_class_indices[i])
+    #         except KeyError:
+    #             print(f"KeyError: Label not found for index {predicted_class_indices[i]}")
+    result = {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}
+    return str(result)
+
+