update README.md

Kawaeee · Feb 10, 2022 · 762a811 · 762a811
1 parent 59940dd
commit 762a811
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -84,7 +84,6 @@
       │   └───valid
       │   │    │   butt/
       │   │    │   bread/
-      │
    ```
 
 ## Model Reproduction
@@ -99,7 +98,7 @@
    pip install -r requirements.txt
    ```
 
- - Run the train.py python script
+ - Run the `train.py` python script
    ```Bash
    python train.py --dataset-path datasets/ --model-path buttbread_resnet152_3.h5
    ```

diff --git a/butt_or_bread/core.py b/butt_or_bread/core.py
diff --git a/butt_or_bread/utils.py b/butt_or_bread/utils.py