fixed requirements for image classifier

x4nth055 · x4nth055 · commit 8d2c5458a12d · 2019-09-14T12:27:34.000+02:00
diff --git a/machine-learning/image-classifier/README.md b/machine-learning/image-classifier/README.md
@@ -2,7 +2,7 @@
 To run this:
 - `pip3 install -r requirements.txt`
 - First, you need to train the model using `python train.py`
-- Edit the code in `test.py` for you optimal model weights in `results` folder ( currently empty because you need to train first ) and run:
+- Edit the code in `test.py` for you optimal model weights in `results` folder ( currently does not because you need to train first ) and run:
     ```
     python test.py
     ```
diff --git a/machine-learning/image-classifier/requirements.txt b/machine-learning/image-classifier/requirements.txt
@@ -1 +1,3 @@
-keras
+keras
+matplotlib
+numpy

-Original file line number
+Diff line change
@@ @@ -1 +1,3 @@ @@
 -keras
 +keras
 +matplotlib
 +numpy