Update app.py
Browse files
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']
|