diff --git a/README.md b/README.md index 332f553..95fd12c 100644 --- a/README.md +++ b/README.md @@ -21,8 +21,19 @@ poetry install ``` -## Running the model +## Download the model ``` -poetry run python ./mol_classifier/classifier.py infer --dir --checkpoint --output +wget https://huggingface.co/ds4sd/MolClassifier/resolve/main/models/molclassifier_model.chpt -P ./data/models/ ``` + +## Running the model + +- Place the images to annotate in `./data/dataset/` + +- Run the model + ``` + poetry run python ./mol_classifier/classifier.py infer --dir ./data/dataset/ --checkpoint ./data/models/molclassifier_model.chpt --output ./data/output.txt + ``` + +- Read predictions in `./data/output.txt` \ No newline at end of file diff --git a/data/dataset/US10525138B2.pdf_11_0.png b/data/dataset/US10525138B2.pdf_11_0.png new file mode 100644 index 0000000..1ecd8f5 Binary files /dev/null and b/data/dataset/US10525138B2.pdf_11_0.png differ diff --git a/data/dataset/US10526356B2.pdf_4_2.png b/data/dataset/US10526356B2.pdf_4_2.png new file mode 100644 index 0000000..51597ad Binary files /dev/null and b/data/dataset/US10526356B2.pdf_4_2.png differ