aditi2222 commited on
Commit
5cf8dd6
·
1 Parent(s): 711cb0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
6
 
7
 
8
  from transformers import pipeline
9
- import gradio as gr
10
  pipe= pipeline('text2text-generation', model="SEBIS/legal_t5_small_trans_fr_en")
11
  def generate_text(inp):
12
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
@@ -38,6 +38,6 @@ def article_summarization(result):
38
 
39
  #iface = gr.Interface(fn=article_summarization,title="Summarization in English",description="facebook/bart-large-cnn for summarization in English", inputs=(lines=50,['text']), outputs=["text"])
40
 
41
- iface = gr.Interface(fn=article_summarization,title="Summarization in English",description="facebook/bart-large-cnn for summarization in English", inputs=gr.inputs.Textbox(lines=50, placeholder="Enter newpaper article to be summarized"), outputs=["text"])
42
 
43
  iface.launch()
 
6
 
7
 
8
  from transformers import pipeline
9
+
10
  pipe= pipeline('text2text-generation', model="SEBIS/legal_t5_small_trans_fr_en")
11
  def generate_text(inp):
12
  output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
 
38
 
39
  #iface = gr.Interface(fn=article_summarization,title="Summarization in English",description="facebook/bart-large-cnn for summarization in English", inputs=(lines=50,['text']), outputs=["text"])
40
 
41
+ iface = gr.Interface(fn=article_summarization,title="Summarization in English",description="facebook/bart-large-cnn for summarization in English", inputs=gr.inputs.Textbox(lines=50, placeholder="Enter newpaper article to be summarized"), outputs=["summary_text"])
42
 
43
  iface.launch()