mfahadkhan commited on
Commit
8bfeedd
·
1 Parent(s): bce7c3d

added for public

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Initialize the summarization pipeline
5
- pipe = pipeline("summarization")
6
 
7
  # Define a function for the summarization
8
  def summarize(text):
@@ -18,4 +18,4 @@ interface = gr.Interface(
18
  )
19
 
20
  # Launch the interface
21
- interface.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Initialize the summarization pipeline with a specified model
5
+ pipe = pipeline("summarization", model="t5-small")
6
 
7
  # Define a function for the summarization
8
  def summarize(text):
 
18
  )
19
 
20
  # Launch the interface
21
+ interface.launch(share=True)