TiberiuCristianLeon commited on
Commit
83386a6
·
verified ·
1 Parent(s): e0efacd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -18,7 +18,9 @@ all_langs = {iso[0]: (iso[1], iso[2], iso[3]) for iso in non_empty_isos} # {'Rom
18
  options = list(favourite_langs.keys())
19
  options.extend(list(all_langs.keys()))
20
 
21
- models = ["Helsinki-NLP", "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul",
 
 
22
  "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-3.3B",
23
  "facebook/mbart-large-50-many-to-many-mmt", "facebook/mbart-large-50-one-to-many-mmt", "facebook/mbart-large-50-many-to-one-mmt",
24
  "facebook/m2m100_418M", "facebook/m2m100_1.2B",
@@ -473,8 +475,7 @@ def swap_languages(src_lang, tgt_lang):
473
  def get_info(model_name, sl, tl):
474
  if model_name == "Helsinki-NLP":
475
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
476
- print(model_name, response)
477
- if 'Repository not found' in response or 'Invalid username or password' in response:
478
  return '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
479
  return response
480
  if model_name == "Argos":
 
18
  options = list(favourite_langs.keys())
19
  options.extend(list(all_langs.keys()))
20
 
21
+ models = ["Helsinki-NLP",
22
+ "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul", "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_nld",
23
+ "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-mul",
24
  "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-3.3B",
25
  "facebook/mbart-large-50-many-to-many-mmt", "facebook/mbart-large-50-one-to-many-mmt", "facebook/mbart-large-50-many-to-one-mmt",
26
  "facebook/m2m100_418M", "facebook/m2m100_1.2B",
 
475
  def get_info(model_name, sl, tl):
476
  if model_name == "Helsinki-NLP":
477
  response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
478
+ if 'Repository not found' in response or 'Invalid username or password' in response.text:
 
479
  return '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
480
  return response
481
  if model_name == "Argos":