updated the path to pretrained model

app.py

@@ -25,7 +25,10 @@ transform_img = transforms.Compose([transforms.ToPILImage(),
 
 
 labels =  ['damage','BG']
-model = Model.load('/Trained_Model.pth', labels) # CHange this while uploading
+ROOT_DIR = os.getcwd()
+path = ROOT_DIR+'/Trained_Model.pth'
+print('path',path)
+model = Model.load(path, labels) # CHange this while uploading