saima730 commited on
Commit
36914a4
·
verified ·
1 Parent(s): f494a3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ models = {
10
  'Urdu': 'Helsinki-NLP/opus-mt-en-ur',
11
  'Arabic': 'Helsinki-NLP/opus-mt-en-ar',
12
  # Add more language models if needed
13
- }
14
 
15
- def load_model(model_name):
16
  """Load the model and tokenizer based on the selected model name."""
17
  model = MarianMTModel.from_pretrained(model_name)
18
  tokenizer = MarianTokenizer.from_pretrained(model_name)
 
10
  'Urdu': 'Helsinki-NLP/opus-mt-en-ur',
11
  'Arabic': 'Helsinki-NLP/opus-mt-en-ar',
12
  # Add more language models if needed
13
+ }
14
 
15
+ def load_model(model_name):
16
  """Load the model and tokenizer based on the selected model name."""
17
  model = MarianMTModel.from_pretrained(model_name)
18
  tokenizer = MarianTokenizer.from_pretrained(model_name)