Fix model loading for CPU-only environments

app.py

@@ -47,7 +47,9 @@ model = SegformerForSemanticSegmentation.from_pretrained(
     label2id=label2id,
     ignore_mismatched_sizes=True,
 )
-loaded_state_dict = torch.load(state_dict_path)
+loaded_state_dict = torch.load(
+    state_dict_path, map_location=torch.device("cpu"), weights_only=True
+)
 model.load_state_dict(loaded_state_dict)
 model = model.to(device)
 model.eval()