mnist : add tensorflow and keras to requirements.txt (#702)

* Added tensorflow and keras to requirements.txt to streamline running MNIST example. Updated instructions to include installing python dependenices for the training script.

* add output example whitespace back

* edit readme and fixed typo

diff --git a/examples/mnist/README.md b/examples/mnist/README.md
index f6b66f74304fdb728eddce3b617f709f3566d298..32e949d4151050b212478a6759a2a361d1c3aaad 100644 (file)
--- a/examples/mnist/README.md
+++ b/examples/mnist/README.md
@@ -3,11 +3,16 @@
 These are simple examples of how to use GGML for inferencing.
 The first example uses convolutional neural network (CNN), the second one uses fully connected neural network.
 
-## Building the examples
+## Python environment setup and build the examples
 
 ```bash
 git clone https://github.com/ggerganov/ggml
 cd ggml
+# Install python dependencies in a virtual environment
+python3 -m venv ggml_env
+source ./ggml_env/bin/activate
+pip install -r requirements.txt
+# Build the examples
 mkdir build && cd build
 cmake ..
 make -j4 mnist-cnn mnist

diff --git a/requirements.txt b/requirements.txt
index a9fd9604e0fe8952276d73859e48178c9178beb5..c686d7fd7c1d5dcf57c397991f831b8de32ca4df 100644 (file)
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,3 +6,5 @@ torchaudio==2.0.2
 torchvision==0.15.2
 transformers==4.29.2
 gguf==0.4.5
+tensorflow==2.15.0
+keras==2.15.0