readme updated

arun477 · Sep 17, 2023 · 28478b9 · 28478b9
1 parent 24e3b46
commit 28478b9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,3 +1,12 @@
+---
+title: Mnist
+emoji: 🐢
+colorFrom: yellow
+colorTo: purple
+sdk: docker
+pinned: false
+---
+
 # <img src="/static/favicon.png" alt="Logo" style="float: left; margin-right: 10px; border-radius:100%;margin-top:5px" />  MNIST CLASSIFIER
 MNIST classifier from scratch
 * Model: CNN

diff --git a/mnist_classifier.ipynb b/mnist_classifier.ipynb
@@ -575,8 +575,7 @@
     "image = TF.to_tensor(image).permute(1, 2, 0)\n",
     "print(image.mean(), image.max(), image.min(), image.std())\n",
     "print(xb[0].mean(), xb[0].max(), xb[0].min(), xb[0].std())\n",
-    "plt.imshow(image);\n",
-    "\n"
+    "plt.imshow(image);"
    ]
   }
  ],