added bounding box

app.py

@@ -20,13 +20,16 @@ def predict(image_url:str = Query(..., description="URL to a chest X-ray image")
         img = transform(img)
         img = torch.from_numpy(img)
         outputs = model(img[None,...])
+        
         prediction = dict(zip(model.pathologies,outputs[0].detach().numpy().tolist()))
+        pred_class=outputs[0].argmax().item()
+        pred_label = model.pathologies[pred_class]
         pred_output = {}
         for k,v in prediction.items():
             pred_output.update({k:round(v,2)})
 
 
-        return {"prediction_result":pred_output}
+        return {"prediction_result":pred_output,"bounding_box":{pred_label:((139,61),(224,132))}}
     except Exception as e:
         print(e)
         raise HTTPException(status_code=400, detail=f"Failed to fetch/process image: {str(e)}")