talha515 commited on
Commit
d529b06
·
verified ·
1 Parent(s): 90760ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ LANGUAGES = {
17
 
18
  # Helper function to load the model and tokenizer
19
  @st.cache_resource
20
- def load_translation_model(src_lanf, tgt_lang):
21
- model_name = f"Helsinki-NLP.opus-mt-{src_lang}-{tgt-lnag}"
22
  tokenizer = MarianTokenizer.from_pretrained(model_name)
23
  model = MarianMTModel.from_pretrained(model_name)
24
  return tokenizer,model
 
17
 
18
  # Helper function to load the model and tokenizer
19
  @st.cache_resource
20
+ def load_translation_model(src_lang, tgt_lang):
21
+ model_name = f"Helsinki-NLP.opus-mt-{src_lang}-{tgt-lang}"
22
  tokenizer = MarianTokenizer.from_pretrained(model_name)
23
  model = MarianMTModel.from_pretrained(model_name)
24
  return tokenizer,model