Update app.py

app.py

@@ -53,6 +53,9 @@ def translate(text: str, model: Optional[str] = MODELS['enro']):
         translation: str = Translate.gemma_direct(text, model)
     return {"input": text, "result": translation, "model": model}
 
+# Keep track of installed (src, tgt) pairs
+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: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'deen'):
@@ -79,7 +82,21 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
         # input_text = [input_text] if isinstance(input_text, str) else input_text           
         config = bergamot.ServiceConfig(numWorkers=4)
         service = bergamot.Service(config)
-        model = service.modelFromConfigPath(f"./{model_name}/bergamot.config.yml")
+        repo_id="TiberiuCristianLeon/Bergamot"
+        branches = ['base', 'base-memory', 'tiny']
+        subfolder = f"{src_code}{tgt_code}"
+            # List all files in the repo
+        all_files = list_repo_files(repo_id, repo_type='model')
+        for branch in branches:
+            branch_files = [f for f in all_files if f.startswith(branch)] 
+            model_files = [f for f in branch_files if f.startswith(model_name)]
+            print(model_files)
+            for file_path in model_files:
+                if subfolder not in file_path:
+                    local_path = hf_hub_download(repo_id=repo_id, filename=file_path)
+                    print(f"Downloaded to: {local_path}")
+        installed_pairs.add(subfolder)
+        model = service.modelFromConfigPath(f"./{model_name}/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