Update README.md

README.md

@@ -239,7 +239,11 @@ for epoch in range(10):
 
     print(f"Epoch {epoch + 1}, Loss: {loss.item():.4f}")
 
-### 5. Evaluate the Model
+```
+
+### Evaluate the Model
+
+
 
 ```python
 
@@ -268,3 +272,4 @@ with torch.no_grad():
 accuracy = correct / total
 
 print(f"Accuracy: {accuracy:.4f}")
+```