PeteBleackley commited on
Commit
3b5865c
1 Parent(s): 4647d83

Fine tuned model was producing nonsense, try mt5

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ import tokenizers
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()
 
11
  import transformers
12
 
13
  pipeline = transformers.pipeline('translation_en_to_ha',
14
+ model='google/mt5-small')
 
15
 
16
  demo = gr.Interface.from_pipeline(pipeline)
17
  demo.launch()