kurianbenoy commited on
Commit
fa3d9d8
1 Parent(s): 1b27acb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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