Sanzana Lora commited on
Commit
81793c7
·
verified ·
1 Parent(s): 28ebfd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- target_language = " "
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',