| from ultralytics import YOLO | |
| from flask import Flask, request, jsonify | |
| app = Flask(__name__) | |
| model = YOLO('best.pt') | |
| def predict(): | |
| image = request.files['file'].read() | |
| results = model(image) | |
| return jsonify(results.pandas().xyxy[0].to_json(orient="records")) | |
| if __name__ == '__main__': | |
| app.run(host='0.0.0.0', port=8080) | |