Sa-m commited on
Commit
153e7c3
·
1 Parent(s): fa7c250

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ app = Flask(__name__)
12
  DETECTION_URL = "/v1/detect"
13
 
14
 
15
- @app.route('/')
16
  def predict():
17
 
18
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True) # force_reload to recache
@@ -27,7 +27,7 @@ def predict():
27
  img = Image.open(io.BytesIO(image_bytes))
28
 
29
  results = model(img, size=640) # reduce size=320 for faster inference
30
- return f"{results.pandas().xyxy[0].to_json(orient="records")}"
31
 
32
 
33
  if __name__ == "__main__":
 
12
  DETECTION_URL = "/v1/detect"
13
 
14
 
15
+ @app.route('/',methods=["POST"]))
16
  def predict():
17
 
18
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True) # force_reload to recache
 
27
  img = Image.open(io.BytesIO(image_bytes))
28
 
29
  results = model(img, size=640) # reduce size=320 for faster inference
30
+ return results.pandas().xyxy[0].to_json(orient="records")
31
 
32
 
33
  if __name__ == "__main__":