Update app.py

app.py

@@ -56,7 +56,7 @@ def get_transforms():
     ])
 
 # Load the model
-model_path = '/content/best_model.pth' 
+model_path = 'model/best_model.pth' 
 model = load_model(model_path)
 transform = get_transforms()