update README

DS4SD · Apr 16, 2024 · dec28a2 · dec28a2
1 parent 19f7271
commit dec28a2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git 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 <your directory containing the images> --checkpoint <path to your checkpoint> --output <your output file name>
+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`
diff --git a/data/dataset/US10525138B2.pdf_11_0.png b/data/dataset/US10525138B2.pdf_11_0.png
diff --git a/data/dataset/US10526356B2.pdf_4_2.png b/data/dataset/US10526356B2.pdf_4_2.png