a fwe changes

main.py

@@ -453,11 +453,11 @@ def predict_image(image_bytes: bytes) -> PredictionResponse:
             success=True,
             predicted_class=disease_info['disease_name'] if disease_info.disease_name else "Unknown",
             predicted_class_index=predicted_class_idx,
-            clean_class_name=clean_name,
+            clean_class_name=disease_info['disease_name'] if disease_info.disease_name else "Unknown",
             message="Prediction successful",
             all_predictions=all_predictions,
             class_id=class_id,
-            label=clean_name,
+            label=class_id,
             confidence=round(confidence, 4),
             confidence_level=confidence_level,
             predictions=all_predictions,