Update app.py

app.py

@@ -4,7 +4,7 @@ from torchvision import transforms
 from PIL import Image
 
 # Load the segmentation model (replace 'path/to/lightmed_model' with the actual path)
-model_path = 'path/to/lightmed_model'
+model_path = 'medsam_lite/lite_medsam.pth'
 segmentation_model = torch.load(model_path, map_location=torch.device('cpu'))
 segmentation_model.eval()