Update main.py

main.py

@@ -81,8 +81,8 @@ def preprocess_image(image_data, img_size=224):
 def predict_image(image_data):
     preprocessed_image = preprocess_image(image_data)
     predictions = model.predict(preprocessed_image)
-    class_idx = np.argmax(predictions, axis=1)[0]
-    confidence = predictions[0][class_idx]
+    class_idx = int(np.argmax(predictions, axis=1)[0])  # Convert to int for JSON serialization
+    confidence = float(predictions[0][class_idx])      # Convert to float for JSON serialization
     class_label = class_labels.get(class_idx, "Unknown")
     if class_label is None:
         return {
@@ -113,4 +113,4 @@ async def api_predict_image(file: UploadFile = File(...)):
 
 # Run the FastAPI app
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)