Spaces:
Runtime error
Runtime error
kurianbenoy
commited on
Commit
•
fa3d9d8
1
Parent(s):
1b27acb
Update app.py
Browse files
app.py
CHANGED
@@ -5,12 +5,10 @@ from fastai.learner import load_learner
|
|
5 |
from transformers import *
|
6 |
import gradio as gr
|
7 |
|
8 |
-
gr.Interface.load("huggingface/kurianbenoy/kde_en_ml_translation_model")
|
9 |
-
|
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
|
|
|
5 |
from transformers import *
|
6 |
import gradio as gr
|
7 |
|
|
|
|
|
8 |
def download():
|
9 |
+
#pretrained_model_name = "Helsinki-NLP/opus-mt-en-ml"
|
10 |
+
#model_cls = AutoModelForSeq2SeqLM
|
11 |
+
#hf_arch, hf_config, hf_tokenizer, hf_model = get_hf_objects(pretrained_model_name, model_cls=model_cls)
|
12 |
|
13 |
model = load_learner(hf_hub_download("kurianbenoy/kde_en_ml_translation_model", "model.pkl"))
|
14 |
return model
|