Update app.py

app.py

@@ -108,7 +108,7 @@ def setup_model():
 # Function to segment image
 def segment_image(image):
   
-    image = cv2.imread(image_path)
+    image = cv2.imread(image)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
     outputs = predictor(image)
@@ -154,14 +154,14 @@ def segment_image(image):
     # Load models
     modernity_model = models.resnet18(pretrained=True)
     modernity_model.fc = nn.Linear(modernity_model.fc.in_features, 5)
-    modernity_checkpoint = torch.load(path +'modernity.pth', map_location=device)
+    modernity_checkpoint = torch.load('modernity.pth', map_location=device)
     modernity_model.load_state_dict(modernity_checkpoint)
     modernity_model.to(device)
     modernity_model.eval()
 
     typicality_model = models.resnet18(pretrained=True)
     typicality_model.fc = nn.Linear(typicality_model.fc.in_features, 5)
-    typicality_checkpoint = torch.load(path + 'typicality.pth', map_location=device)
+    typicality_checkpoint = torch.load('typicality.pth', map_location=device)
     typicality_model.load_state_dict(typicality_checkpoint)
     typicality_model.to(device)
     typicality_model.eval()