Spaces:
Sleeping
Sleeping
stestoni91
commited on
Commit
•
d459b8e
1
Parent(s):
4738e5b
Change main function name
Browse files- app.py +2 -2
- test_app.py +3 -3
app.py
CHANGED
@@ -3,13 +3,13 @@ import gradio as gr
|
|
3 |
|
4 |
model = pipeline('summarization', model='facebook/bart-large-cnn')
|
5 |
|
6 |
-
def
|
7 |
summary = model(text)[0]['summary_text']
|
8 |
return summary
|
9 |
|
10 |
with gr.Blocks() as ui:
|
11 |
textbox = gr.Textbox(placeholder='Enter text to summarize', label='input')
|
12 |
-
gr.Interface(fn=
|
13 |
|
14 |
if __name__ == '__main__':
|
15 |
ui.launch()
|
|
|
3 |
|
4 |
model = pipeline('summarization', model='facebook/bart-large-cnn')
|
5 |
|
6 |
+
def summarize(text):
|
7 |
summary = model(text)[0]['summary_text']
|
8 |
return summary
|
9 |
|
10 |
with gr.Blocks() as ui:
|
11 |
textbox = gr.Textbox(placeholder='Enter text to summarize', label='input')
|
12 |
+
gr.Interface(fn=summarize, inputs=textbox, outputs='text')
|
13 |
|
14 |
if __name__ == '__main__':
|
15 |
ui.launch()
|
test_app.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
-
from app import
|
2 |
import wikipedia
|
3 |
wikipedia.set_lang("it")
|
4 |
import warnings
|
5 |
warnings.filterwarnings("ignore")
|
6 |
|
7 |
|
8 |
-
def
|
9 |
text = wikipedia.summary('Sardegna', sentences=10)
|
10 |
-
summary =
|
11 |
assert len(text) > len(summary)
|
|
|
1 |
+
from app import summarize
|
2 |
import wikipedia
|
3 |
wikipedia.set_lang("it")
|
4 |
import warnings
|
5 |
warnings.filterwarnings("ignore")
|
6 |
|
7 |
|
8 |
+
def test_summarize():
|
9 |
text = wikipedia.summary('Sardegna', sentences=10)
|
10 |
+
summary = summarize(text)
|
11 |
assert len(text) > len(summary)
|