Spaces:
Runtime error
Runtime error
kurianbenoy
commited on
Commit
•
865e5dd
1
Parent(s):
4ac6c50
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ gr.Interface.load("huggingface/kurianbenoy/kde_en_ml_translation_model")
|
|
10 |
def download():
|
11 |
pretrained_model_name = "Helsinki-NLP/opus-mt-en-ml"
|
12 |
model_cls = AutoModelForSeq2SeqLM
|
13 |
-
hf_arch, hf_config, hf_tokenizer, hf_model =
|
14 |
|
15 |
model = load_learner(hf_hub_download("kurianbenoy/kde_en_ml_translation_model", "model.pkl"))
|
16 |
return model
|
|
|
10 |
def download():
|
11 |
pretrained_model_name = "Helsinki-NLP/opus-mt-en-ml"
|
12 |
model_cls = AutoModelForSeq2SeqLM
|
13 |
+
hf_arch, hf_config, hf_tokenizer, hf_model = get_hf_objects(pretrained_model_name, model_cls=model_cls)
|
14 |
|
15 |
model = load_learner(hf_hub_download("kurianbenoy/kde_en_ml_translation_model", "model.pkl"))
|
16 |
return model
|