Update README.md

README.md

@@ -18,12 +18,15 @@ To use the models, please follow the steps below.
 Prerequisites
 The following Python libraries are required to load and run the models:
 
+```
 pip install torch scikit-learn joblib huggingface-hub
+```
 
 
 Loading the Models
 The following script allows you to download the model files from this repository and load them into memory for later use.
 
+```
 import torch
 import torch.nn as nn
 from joblib import load
@@ -82,6 +85,7 @@ rf_prediction = rf_model_loaded.predict(sample_image_flat)
 
 print(f"Prediction by the Neural Network: {ffnn_prediction.item()}")
 print(f"Prediction by the Random Forest: {rf_prediction[0]}")
+```
 
 Notes
 The SimpleFFNN class must be defined to allow the PyTorch model to be loaded.