Update install_model.py

install_model.py

@@ -1,37 +1,10 @@
-import os
-import zipfile
 import argostranslate.package
-import shutil
-
-model_dir = "/app/models"
-install_dir = "/app/.local/share/argos-translate/packages"
-
-# Cleanup: Remove invalid model folder if it exists
-bad_path = os.path.join(install_dir, "model")
-if os.path.isdir(bad_path):
-    shutil.rmtree(bad_path)
-    print(f"Removed invalid leftover package: {bad_path}")
-
-def is_valid_argosmodel(path):
-    try:
-        with zipfile.ZipFile(path, 'r') as archive:
-            return 'metadata.json' in archive.namelist()
-    except zipfile.BadZipFile:
-        return False
-
-# Install all valid .argosmodel files
-for filename in os.listdir(model_dir):
-    if filename.endswith(".argosmodel"):
-        full_path = os.path.join(model_dir, filename)
-
-        if not is_valid_argosmodel(full_path):
-            print(f"Skipping invalid model file: {filename}")
-            continue
-
-        try:
-            print(f"Installing model: {filename}")
-            argostranslate.package.install_from_path(full_path)
-        except Exception as e:
-            print(f"Error installing {filename}: {e}")
+import os
 
-print("All valid models installed.")
+model_path = "/app/models/en_kab_comp"
+if os.path.isdir(model_path):
+    package = argostranslate.package.Package(model_path)
+    argostranslate.package.install_from_path(package)
+    print("Kabyle model installed.")
+else:
+    print("Model path not found.")