haydpw commited on
Commit
cf104bc
·
1 Parent(s): 0d3ea05

change original image response

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -85,7 +85,7 @@ async def predict_image(file: UploadFile = File(...)):
85
  combined_image_base64 = image_to_base64(combined_image, "PNG")
86
 
87
  response = {
88
- "original_image": image_to_base64(image_bg_removed),
89
  "segmentation_results":results,
90
  "combined_image": combined_image_base64
91
  }
 
85
  combined_image_base64 = image_to_base64(combined_image, "PNG")
86
 
87
  response = {
88
+ "original_image": image_to_base64(pil_image),
89
  "segmentation_results":results,
90
  "combined_image": combined_image_base64
91
  }