Update app.py

app.py

@@ -7,7 +7,7 @@ from train import UNet
 import numpy as np 
 
 # Load the trained model
-model_path = '/model.pth'
+model_path = 'model.pth'
 model = UNet(n_channels=3, n_classes=6)
 model.load_state_dict(torch.load(model_path))
 model.eval()