Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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)
|