Spaces:
Sleeping
Sleeping
Changed error message to warning messages.
Browse files- app.py +1 -1
- controlled_summarization.py +8 -4
app.py
CHANGED
@@ -185,4 +185,4 @@ with gr.Blocks(css="#htext span {white-space: pre-line}") as demo:
|
|
185 |
)
|
186 |
|
187 |
|
188 |
-
demo.launch(share=False)
|
|
|
185 |
)
|
186 |
|
187 |
|
188 |
+
demo.queue().launch(share=False)
|
controlled_summarization.py
CHANGED
@@ -108,7 +108,8 @@ def ctrlsum_for_str(input, length=None, keywords=None) -> List[Tuple[str, str]]:
|
|
108 |
def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> List[Tuple[str, str, str]]:
|
109 |
if input == None and url == "":
|
110 |
if text == "":
|
111 |
-
|
|
|
112 |
else:
|
113 |
return ctrlsum_for_str(text, length, keywords), text, None
|
114 |
else:
|
@@ -133,7 +134,8 @@ def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> L
|
|
133 |
|
134 |
filename = download_pdf(url, './cache/')
|
135 |
else:
|
136 |
-
|
|
|
137 |
else:
|
138 |
filename = input.name
|
139 |
if keywords != "":
|
@@ -151,13 +153,15 @@ def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> L
|
|
151 |
results = ctrlsum_pipeline.predict(filename,
|
152 |
save_results=False, length=length, keywords=keywords, num_beams=1)
|
153 |
else:
|
154 |
-
|
|
|
155 |
|
156 |
output = []
|
157 |
for res in results["summary"]:
|
158 |
output.append(f"{res}\n\n")
|
159 |
if results["raw_text"] == "":
|
160 |
-
|
|
|
161 |
return "".join(output), results["raw_text"], filename
|
162 |
|
163 |
|
|
|
108 |
def ctrlsum_for_file(input=None, length=None, keywords="", text="", url="") -> List[Tuple[str, str, str]]:
|
109 |
if input == None and url == "":
|
110 |
if text == "":
|
111 |
+
gr.Warning("Input cannot be left blank!")
|
112 |
+
return None, None, None
|
113 |
else:
|
114 |
return ctrlsum_for_str(text, length, keywords), text, None
|
115 |
else:
|
|
|
134 |
|
135 |
filename = download_pdf(url, './cache/')
|
136 |
else:
|
137 |
+
gr.Warning("Invalid URL (Not PDF)!")
|
138 |
+
return None, None, None
|
139 |
else:
|
140 |
filename = input.name
|
141 |
if keywords != "":
|
|
|
153 |
results = ctrlsum_pipeline.predict(filename,
|
154 |
save_results=False, length=length, keywords=keywords, num_beams=1)
|
155 |
else:
|
156 |
+
gr.Warning("File Format Error! Please upload .txt or .pdf files.")
|
157 |
+
return None, None, None
|
158 |
|
159 |
output = []
|
160 |
for res in results["summary"]:
|
161 |
output.append(f"{res}\n\n")
|
162 |
if results["raw_text"] == "":
|
163 |
+
gr.Warning("Unable to parse File! Please try a different file.")
|
164 |
+
return None, None, None
|
165 |
return "".join(output), results["raw_text"], filename
|
166 |
|
167 |
|