Modify import code

README.md

@@ -15,13 +15,20 @@ where the data came from and in which folder it must go.
 ## Neccessary Package
   - Since the model all runs on Hugging Face, you don't need to install any packages.
   - However, if you do want to run the model locally, you will need to install these following packages:
-    ``import torch``
-      ``import torch.nn as nn
+    ``import torch
+    
+      import torch.nn as nn
+    
       import torch.optim as optim
+    
       import torch.nn.functional as F
+    
       import numpy as np
+    
       import torchvision
+    
       from torchvision import *
+    
       from torch.utils.data import Dataset, DataLoader``
 
 ## Running