Update app.py

app.py

@@ -126,11 +126,13 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
             for file_path in model_files:
                 if subfolder not in installed_pairs:
                     local_path = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename=file_path, local_dir=subfolder)
-                    print(f"Downloaded to: {local_path}")
-                    localfolder = local_path.split(f"/model_name")[0]
+                    print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
+                    localfolder = local_path.split(f"{subfolder}/{model_name}")[0]
+                else:
+                    localfolder = f"{subfolder}/{model_name}"
         installed_pairs.add(subfolder)
         print('installed_pairs', installed_pairs, 'localfolder', localfolder)
-        model = service.modelFromConfigPath(f"{localfolder}/{model_name}/config.yml")
+        model = service.modelFromConfigPath(f"{localfolder}/config.yml")
         options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
         rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
         response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text