ychenNLP commited on
Commit
c4afd50
1 Parent(s): b53a718

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -214,7 +214,7 @@ Zulu zul_Latn'''
214
  def load_models():
215
  # build model and tokenizer
216
  model_name_dict = {
217
- 'nllb-3.3B': "ychenNLP/nllb-200-3.3b-ep",
218
  }
219
 
220
  model_dict = {}
@@ -231,7 +231,7 @@ def load_models():
231
 
232
  def translation(source, target, text):
233
  if len(model_dict) == 2:
234
- model_name = 'nllb-3.3B'
235
 
236
  start_time = time.time()
237
  source = flores_codes[source]
@@ -280,7 +280,7 @@ if __name__ == '__main__':
280
 
281
  title = "EasyProject: a simple end-to-end label projection method for cross-lingual transfer"
282
 
283
- demo_status = "Check out our paper: Frustratingly Easy Label Projection for Cross-lingual Transfer (https://arxiv.org/abs/2211.15613). Powered by NLLB-200-3.3B (ychenNLP/nllb-200-3.3b-ep)."
284
  description = f"{demo_status}"
285
  examples = [
286
  ['English', 'Akan', "Davies is leaving to become chairman of the [London School of Economics]"],
 
214
  def load_models():
215
  # build model and tokenizer
216
  model_name_dict = {
217
+ 'nllb-1.3B': "ychenNLP/nllb-200-distilled-1.3B-easyproject",
218
  }
219
 
220
  model_dict = {}
 
231
 
232
  def translation(source, target, text):
233
  if len(model_dict) == 2:
234
+ model_name = 'nllb-1.3B'
235
 
236
  start_time = time.time()
237
  source = flores_codes[source]
 
280
 
281
  title = "EasyProject: a simple end-to-end label projection method for cross-lingual transfer"
282
 
283
+ demo_status = "Check out our paper: Frustratingly Easy Label Projection for Cross-lingual Transfer (https://arxiv.org/abs/2211.15613). Powered by (ychenNLP/nllb-200-distilled-1.3B-easyproject)."
284
  description = f"{demo_status}"
285
  examples = [
286
  ['English', 'Akan', "Davies is leaving to become chairman of the [London School of Economics]"],