Spaces:
Sleeping
Sleeping
Commit
·
8bfeedd
1
Parent(s):
bce7c3d
added for public
Browse files
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)
|