d7ec164171ea9b9739b6827bf0c3d328a9c0c564,demo.py,,,#,39

Before Change

if __name__ == "__main__":

    // If the model has been trained load the model, otherwise use the given model
    export_models_dir = glob("demo/page_model/export/*")
    if not export_models_dir:
        model_dir = "demo/model/"
    else:
        export_models_dir.sort()

After Change

    // If the model has been trained load the model, otherwise use the given model
    model_dir = "demo/page_model/export"
    if not os.path.exists(model_dir):
        model_dir = "demo/model/"

    input_files = glob("demo/pages/test_a1/images/*")

    output_dir = "demo/processed_images"
    os.makedirs(output_dir, exist_ok=True)

Instances