update app.py

app.py

@@ -36,13 +36,9 @@ from torchvision import datasets, transforms, utils
 
 save1_or_load0 = False
 
-!mkdir -p /content # This path will be used for saving the weights file after training
+if not os.path.exists("/content"): 
+    os.makedirs("/content") # This path will be used for saving the weights file after training
 
-!git clone https://github.com/rajayourfriend/EraV2/
-!cp EraV2/S14/*.py .
-!cp EraV2/S14/*.jpg .
-if save1_or_load0 == False:
-  !cp /content/EraV2/S14/weights_92.ckpt /content/weights.ckpt
 wt_fname = "/content/weights.ckpt" # weights file name to load
 
 model, trainer = ts_lt(save1_or_load0, Epochs = 26, wt_fname = "/content/weights.ckpt") # Train and Save Vs Load and Test