Update README.md

README.md

@@ -16,7 +16,7 @@ This model is extremely weak. I am not good at data science
 
 <details>
 <summary><b>Trained on 500 Epoch with 2.1 million song data from Spotify Database</b></summary>
-```python
+```
 import torch
 import torch.nn as nn
 import torch.optim as optim
@@ -87,4 +87,5 @@ model.eval()
 with torch.no_grad():
     predicted = model(X_test_tensor)
 ```
+
 </details>