Hussein El-Hadidy commited on
Commit
ba736d8
·
1 Parent(s): 4c67faf

Changed Json

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. main.py +3 -2
app.py CHANGED
@@ -229,8 +229,9 @@ async def process_image(file: UploadFile = File(...)):
229
  show=False, # Set to False for API use
230
  save=False
231
  )
232
- result_json = results[0].tojson()
 
233
  except Exception as e:
234
  raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
235
 
236
- return JSONResponse(content={"message": "Image processed successfully", "results": result_json})
 
229
  show=False, # Set to False for API use
230
  save=False
231
  )
232
+ resultsKeypointsXY = results[0].keypoints.xy
233
+
234
  except Exception as e:
235
  raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
236
 
237
+ return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY})
main.py CHANGED
@@ -229,8 +229,9 @@ async def process_image(file: UploadFile = File(...)):
229
  show=False, # Set to False for API use
230
  save=False
231
  )
232
- result_json = results[0].tojson()
 
233
  except Exception as e:
234
  raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
235
 
236
- return JSONResponse(content={"message": "Image processed successfully", "results": result_json})
 
229
  show=False, # Set to False for API use
230
  save=False
231
  )
232
+ resultsKeypointsXY = results[0].keypoints.xy
233
+
234
  except Exception as e:
235
  raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
236
 
237
+ return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY})