Hwilner commited on
Commit
ac9bb28
·
verified ·
1 Parent(s): bc233ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,8 +4,8 @@ import gradio as gr
4
  # Use a pipeline as a high-level helper
5
  from transformers import pipeline
6
 
7
- text_summary = pipeline("summarization", model="dicta-il/dictalm2.0", torch_dtype=torch.bfloat16)
8
-
9
 
10
 
11
 
@@ -29,5 +29,5 @@ demo = gr.Interface(fn=summary,
29
  inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
30
  outputs=[gr.Textbox(label="Summarized text",lines=4)],
31
  title="Hebrew Text Summarizer",
32
- description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE TEXT")
33
  demo.launch()
 
4
  # Use a pipeline as a high-level helper
5
  from transformers import pipeline
6
 
7
+ # text_summary = pipeline("summarization", model="dicta-il/dictalm2.0", torch_dtype=torch.bfloat16)
8
+ text_summary = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum", torch_dtype=torch.bfloat16)
9
 
10
 
11
 
 
29
  inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
30
  outputs=[gr.Textbox(label="Summarized text",lines=4)],
31
  title="Hebrew Text Summarizer",
32
+ description="This application will summarize Hebrew text.")
33
  demo.launch()