TestAPI / app /__init__.py
pawandev
Added new pan model and change ocrsetting and extraction regex as per new model
43ca488
raw
history blame
415 Bytes
from flask import Flask
from ultralytics import YOLO
def create_app():
app = Flask(__name__)
from .api import ocr_bp
app.register_blueprint(ocr_bp)
with app.app_context():
# Load model once
app.models = {
'adhaarModel': YOLO('models/aadhaarYolov8.pt'),
'panModel': YOLO('models/PanModel_v5.pt') # Load additional models as needed
}
return app