Upload README.md with huggingface_hub

README.md

@@ -46,7 +46,7 @@ cd tiny-cnn-classifier
 ### 2. Install Dependencies
 ```bash
 pip install torch torchvision pillow
-
+```
 ### 3. Load the model
 ```python 
 import torch
@@ -56,7 +56,7 @@ from model import CustomCNN
 model = CustomCNN()
 model.load_state_dict(torch.load("cnn_model.pth", map_location="cpu"))
 model.eval()
-
+```
 ### 4. Run inference on an image
 ```python
 from torchvision import transforms
@@ -79,7 +79,7 @@ with torch.no_grad():
 
 classes = ['airplane','automobile','bird','cat','deer','dog','frog','horse','ship','truck']
 print("Predicted class:", classes[pred_class])
-
+```
 
 ### Training Information