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',
|