Update main.py

main.py

@@ -27,12 +27,12 @@ class_labels = {
     3: "Apple___healthy",
     4: "Background_without_leaves",
     5: "Blueberry___healthy",
-    6: "Cherry_(including_sour)___Powdery_mildew",
-    7: "Cherry_(including_sour)___healthy",
-    8: "Corn_(maize)___Cercospora_leaf_spot Gray_leaf_spot",
-    9: "Corn_(maize)___Common_rust_",
-    10: "Corn_(maize)___Northern_Leaf_Blight",
-    11: "Corn_(maize)___healthy",
+    6: "Cherry___Powdery_mildew",
+    7: "Cherry___healthy",
+    8: "Corn___Cercospora_leaf_spot Gray_leaf_spot",
+    9: "Corn___Common_rust_",
+    10: "Corn___Northern_Leaf_Blight",
+    11: "Corn___healthy",
     12: "Grape___Black_rot",
     13: "Grape___Esca_(Black_Measles)",
     14: "Grape___Leaf_blight_(Isariopsis_Leaf_Spot)",
@@ -93,12 +93,8 @@ def predict_image(image_data):
     except Exception as e:
         external_data = {"error": str(e)}
 
-    return {
-        "class_index": class_idx,
-        "class_label": class_label,
-        "confidence": confidence,
-        "external_data": external_data
-    }
+    return external_data
+    
 
 # Route for health check
 @app.get("/health")