complete the notebook

truefoundry · Feb 19, 2024 · 179f862 · 179f862
1 parent 9b38440
commit 179f862
Show file tree

Hide file tree

Showing 3 changed files with 337 additions and 2 deletions.
diff --git a/mnist-classifaction/deploy_model.ipynb b/mnist-classifaction/deploy_model.ipynb
diff --git a/mnist-classifaction/deploy_model/fastapi_service.py b/mnist-classifaction/deploy_model/fastapi_service.py
@@ -18,7 +18,6 @@ def load_image(img_url):
     img_path = tf.keras.utils.get_file("image.jpg", img_url)
     img = tf.keras.preprocessing.image.load_img(img_path, target_size=(28, 28))
     img_arr = tf.keras.preprocessing.image.img_to_array(img)
-    img_arr = tf.image.rgb_to_grayscale(img_arr) / 255.0
     return img_arr 
 
 

diff --git a/mnist-classifaction/train_model.ipynb b/mnist-classifaction/train_model.ipynb
@@ -6,7 +6,7 @@
     "id": "QOAnoPl-dlSY"
    },
    "source": [
-    "# Train and Deploy Model on Truefoundry\n",
+    "# Train Model on Truefoundry\n",
     "This notebook demonstrates a demo on how you can train an image classification model on mnist dataset and deploy the model training job on truefoundry platform."
    ]
   },