richardorama
commited on
Commit
β’
9e19736
1
Parent(s):
8cf4aa8
Update app.py
Browse files
app.py
CHANGED
@@ -18,10 +18,10 @@ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
|
18 |
# 2010 marriage license application, according to court documents.
|
19 |
# """
|
20 |
|
21 |
-
|
22 |
|
23 |
-
# Create a text area for user input
|
24 |
-
|
25 |
|
26 |
# Define the summarization function
|
27 |
def summarize(txt):
|
@@ -31,11 +31,25 @@ def summarize(txt):
|
|
31 |
summary = summarizer(txt, max_length=130, min_length=30, do_sample=False)
|
32 |
st.write(summary[0]['summary_text'])
|
33 |
|
34 |
-
|
35 |
-
|
36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
else:
|
38 |
-
st.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
|
40 |
|
41 |
|
|
|
18 |
# 2010 marriage license application, according to court documents.
|
19 |
# """
|
20 |
|
21 |
+
# DEFAULT_STATEMENT = ""
|
22 |
|
23 |
+
# # Create a text area for user input
|
24 |
+
# STATEMENT = st.sidebar.text_area('Enter Article (String)', DEFAULT_ARTICLE, height=150)
|
25 |
|
26 |
# Define the summarization function
|
27 |
def summarize(txt):
|
|
|
31 |
summary = summarizer(txt, max_length=130, min_length=30, do_sample=False)
|
32 |
st.write(summary[0]['summary_text'])
|
33 |
|
34 |
+
DEFAULT_STATEMENT = ""
|
35 |
+
# Create a text area for user input
|
36 |
+
STATEMENT = st.sidebar.text_area('Enter Statement (String or List of Strings)', DEFAULT_STATEMENT, height=150)
|
37 |
+
|
38 |
+
# Enable the button only if there is text in the SENTIMENT variable
|
39 |
+
if STATEMENT:
|
40 |
+
if st.sidebar.button('Summarize Statement'):
|
41 |
+
# Call your Summarize function here
|
42 |
+
#st.write(f"Summarizing: {STATEMENT}")
|
43 |
+
summarize(STATEMENT) # Directly pass the STATEMENT
|
44 |
else:
|
45 |
+
st.sidebar.button('Summarize Statement', disabled=True)
|
46 |
+
st.warning('π Please enter Statement!')
|
47 |
+
|
48 |
+
# # Create a button and trigger the summarize function when clicked
|
49 |
+
# if st.sidebar.button('Summarize Article'):
|
50 |
+
# summarize(ARTICLE)
|
51 |
+
# else:
|
52 |
+
# st.warning('π Please enter Article!')
|
53 |
|
54 |
|
55 |
|