gsarti commited on
Commit
7f42f0d
1 Parent(s): e3beb4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import gradio.inputs
3
  import gradio.outputs
4
  from transformers.pipelines import pipeline
5
 
6
- model_names = ["IT5 Small", "IT5 Base", "IT5 Large", "mT5 Small", "mT5 Base"]
7
- model_values = ['it5-small', 'it5-base', 'it5-large', 'mt5-small', 'mt5-base']
8
 
9
  task_names = ["Headline Generation", "Question Answering", "Question Generation", "Wikipedia Summarization", "News Summarization", "Formal to Informal Style Transfer", "Informal to Formal Style Transfer", "Il Giornale to Repubblica Headline Style Transfer", "Repubblica to Il Giornale Headline Style Transfer"]
10
  task_values = ['headline-generation', 'question-answering', 'question-generation', 'wiki-summarization', 'news-summarization', 'formal-to-informal', 'informal-to-formal', 'ilgiornale-to-repubblica', 'repubblica-to-ilgiornale']
 
3
  import gradio.outputs
4
  from transformers.pipelines import pipeline
5
 
6
+ model_names = ["IT5 Small", "IT5 Base", "IT5 Large", "mT5 Small", "mT5 Base", "IT5 Small Efficient"]
7
+ model_values = ['it5-small', 'it5-base', 'it5-large', 'mt5-small', 'mt5-base', "it5-efficient-small-el32"]
8
 
9
  task_names = ["Headline Generation", "Question Answering", "Question Generation", "Wikipedia Summarization", "News Summarization", "Formal to Informal Style Transfer", "Informal to Formal Style Transfer", "Il Giornale to Repubblica Headline Style Transfer", "Repubblica to Il Giornale Headline Style Transfer"]
10
  task_values = ['headline-generation', 'question-answering', 'question-generation', 'wiki-summarization', 'news-summarization', 'formal-to-informal', 'informal-to-formal', 'ilgiornale-to-repubblica', 'repubblica-to-ilgiornale']