Hugging Face's logo

Spaces:
themasterbetters
/
the-master-betters-translator
Running

App Files Community
dipesh1701 commited on
Commit
5c80f7d
•
1 Parent(s): d0626da

add bigscience/bloom

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ from flores200_codes import flores_codes
9
  def load_models():
10
  # build model and tokenizer
11
  model_name_dict = {
12
- "nllb-distilled-600M": "facebook/nllb-200-distilled-600M",
 
13
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
14
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
15
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
 
9
  def load_models():
10
  # build model and tokenizer
11
  model_name_dict = {
12
+ "bloom": "bigscience/bloom",
13
+ #"nllb-distilled-600M": "facebook/nllb-200-distilled-600M",
14
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
15
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
16
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',