Update app.py

app.py

@@ -87,7 +87,7 @@ installed_pairs = set()
 
 # https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
 @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
-def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'deen'):
+def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
     """
     Translates the input text from the source language to the target language using a specified model.
     Parameters:
@@ -127,7 +127,7 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
                 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}") # Downloaded to: deen/base/deen/config.yml
-                    localfolder = local_path.split(f"{subfolder}/{model_name}")[0]
+                    localfolder = local_path.rsplit('/', 1)[0]
                 else:
                     localfolder = f"{subfolder}/{model_name}"
         installed_pairs.add(subfolder)