Sanzana Lora commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,8 +4,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
| 4 |
|
| 5 |
|
| 6 |
# Function for cross-lingual summarization
|
| 7 |
-
def cross_lingual_summarization(article_text):
|
| 8 |
-
|
| 9 |
WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))
|
| 10 |
|
| 11 |
# Load the model and tokenizer
|
|
@@ -55,7 +55,7 @@ iface = gr.Interface(
|
|
| 55 |
fn=cross_lingual_summarization,
|
| 56 |
inputs=[
|
| 57 |
gr.Textbox(label='Article Text'),
|
| 58 |
-
gr.Dropdown(['amharic', 'arabic', 'azerbaijani', 'bengali', 'burmese', 'chinese_simplified', 'chinese_traditional',
|
| 59 |
'english', 'french', 'gujarati', 'hausa', 'hindi', 'igbo', 'indonesian', 'japanese', 'kirundi',
|
| 60 |
'korean', 'kyrgyz', 'marathi', 'nepali', 'oromo', 'pashto', 'persian', 'pidgin', 'portuguese',
|
| 61 |
'punjabi', 'russian', 'scottish_gaelic', 'serbian_cyrillic', 'serbian_latin', 'sinhala', 'somali',
|
|
|
|
| 4 |
|
| 5 |
|
| 6 |
# Function for cross-lingual summarization
|
| 7 |
+
def cross_lingual_summarization(article_text, target_language):
|
| 8 |
+
|
| 9 |
WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))
|
| 10 |
|
| 11 |
# Load the model and tokenizer
|
|
|
|
| 55 |
fn=cross_lingual_summarization,
|
| 56 |
inputs=[
|
| 57 |
gr.Textbox(label='Article Text'),
|
| 58 |
+
gr.Dropdown(choices = ['amharic', 'arabic', 'azerbaijani', 'bengali', 'burmese', 'chinese_simplified', 'chinese_traditional',
|
| 59 |
'english', 'french', 'gujarati', 'hausa', 'hindi', 'igbo', 'indonesian', 'japanese', 'kirundi',
|
| 60 |
'korean', 'kyrgyz', 'marathi', 'nepali', 'oromo', 'pashto', 'persian', 'pidgin', 'portuguese',
|
| 61 |
'punjabi', 'russian', 'scottish_gaelic', 'serbian_cyrillic', 'serbian_latin', 'sinhala', 'somali',
|