Update README.md

README.md

@@ -29,7 +29,7 @@ model-index:
 
 Use the code below to get started with the model.
 
-```
+```python
 model = MixtureOfExperts(num_experts=10)
 
 checkpoint_path = "FP_ML_MOE_SIMPLE_99_75.pth"
@@ -70,18 +70,9 @@ SGD with learning rate of 0.01 and learning rate decay to 0.001
 https://huggingface.co/datasets/ylecun/mnist
 
 #### Metrics
-model-index:
-  - name: MoE-CNN
-    results:
-      - task:
-          type: image-classification
-        dataset:
-          name: MNIST
-          type: image-classification
-        metrics:
-          - name: Accuracy
-            type: Accuracy
-            value: 99.75
+Accuracy: 99.75%
+Error rate: 0.25%
+
 ## Technical Specifications [optional]
 
 ### Model Architecture