PeteBleackley commited on
Commit
c5af2f5
1 Parent(s): 7cce974

New t5 based model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ import tokenizers
11
  import transformers
12
 
13
  pipeline = transformers.pipeline('translation_en_to_ha',
14
- model='google/madlad400-3b-mt')
 
15
 
16
  demo = gr.Interface.from_pipeline(pipeline)
17
  demo.launch()
 
11
  import transformers
12
 
13
  pipeline = transformers.pipeline('translation_en_to_ha',
14
+ model='PeteBleackley/t5-small-eng-hau',
15
+ tokenizer='google-t5/t5-small')
16
 
17
  demo = gr.Interface.from_pipeline(pipeline)
18
  demo.launch()