diff --git a/medcat-v2/medcat/utils/legacy/conversion_all.py b/medcat-v2/medcat/utils/legacy/conversion_all.py index 5bde9f06b..7b25768a6 100644 --- a/medcat-v2/medcat/utils/legacy/conversion_all.py +++ b/medcat-v2/medcat/utils/legacy/conversion_all.py @@ -31,9 +31,12 @@ def __init__(self, medcat1_model_pack_path: str, ser_type: AvailableSerialisers = AvailableSerialisers.dill): if medcat1_model_pack_path.endswith(".zip"): folder_path = medcat1_model_pack_path[:-4] - self.old_model_name = os.path.split(medcat1_model_pack_path)[1] + self.old_model_name = os.path.split( + medcat1_model_pack_path)[1].rsplit(".zip", 1)[0] unpack(medcat1_model_pack_path, folder_path) medcat1_model_pack_path = folder_path + else: + self.old_model_name = os.path.split(medcat1_model_pack_path)[1] if not os.path.isdir(medcat1_model_pack_path): raise ValueError( "Provided model path is not a directory: "