Kleber commited on
Commit
06cb1ab
1 Parent(s): ddb72fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def translate(text, source_lang, target_lang, max_length=400):
40
  # return "Error: the target language is incorrect"
41
 
42
  translation_pipeline = pipeline(TASK,
43
- model=model,
44
  tokenizer=tokenizer,
45
  src_lang=source_lang,
46
  tgt_lang=target_lang,
 
40
  # return "Error: the target language is incorrect"
41
 
42
  translation_pipeline = pipeline(TASK,
43
+ model=general_model,
44
  tokenizer=tokenizer,
45
  src_lang=source_lang,
46
  tgt_lang=target_lang,