Update app.py

app.py

@@ -35,12 +35,7 @@ from torchvision import datasets, transforms, utils
 
 save1_or_load0 = False
 
-if not os.path.exists("/content"): 
-    os.makedirs("/content") # This path will be used for saving the weights file after training
-
-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
+model, trainer = ts_lt(save1_or_load0, Epochs = 26, wt_fname = "weights.ckpt") # Train and Save Vs Load and Test
 '''
 ts_lt(save1_or_load0, # decision maker for training Vs testing
           Epochs = 1, # argument for training