Spaces:
Sleeping
Sleeping
Commit
•
7bae3b7
1
Parent(s):
5100407
updated model to google-t5/t5-small
Browse files
app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
-
|
4 |
|
5 |
# Function to perform summarization
|
6 |
def summarize_text(text):
|
7 |
try:
|
8 |
-
summerizer = pipeline('summarization', model='google/t5-small')
|
9 |
summarized_text = summerizer(text)[0]['summary_text']
|
10 |
return summarized_text
|
11 |
except Exception as e:
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
+
#google-t5/t5-small
|
4 |
|
5 |
# Function to perform summarization
|
6 |
def summarize_text(text):
|
7 |
try:
|
8 |
+
summerizer = pipeline('summarization', model='google-t5/t5-small')
|
9 |
summarized_text = summerizer(text)[0]['summary_text']
|
10 |
return summarized_text
|
11 |
except Exception as e:
|