kkastr
commited on
Commit
•
e3c8b51
1
Parent(s):
77d9839
forgot to return figure
Browse files
app.py
CHANGED
|
@@ -136,6 +136,7 @@ def summarizer(url: str) -> str:
|
|
| 136 |
plt.imshow(wcloud)
|
| 137 |
plt.axis("off")
|
| 138 |
plt.tight_layout()
|
|
|
|
| 139 |
lst_summaries = []
|
| 140 |
|
| 141 |
nlp = pipeline('summarization', model="sshleifer/distilbart-cnn-12-6")
|
|
@@ -153,7 +154,7 @@ def summarizer(url: str) -> str:
|
|
| 153 |
|
| 154 |
long_output = submission_title + '\n' + '\n' + joined_summaries
|
| 155 |
|
| 156 |
-
return short_output, long_output
|
| 157 |
|
| 158 |
|
| 159 |
if __name__ == "__main__":
|
|
|
|
| 136 |
plt.imshow(wcloud)
|
| 137 |
plt.axis("off")
|
| 138 |
plt.tight_layout()
|
| 139 |
+
|
| 140 |
lst_summaries = []
|
| 141 |
|
| 142 |
nlp = pipeline('summarization', model="sshleifer/distilbart-cnn-12-6")
|
|
|
|
| 154 |
|
| 155 |
long_output = submission_title + '\n' + '\n' + joined_summaries
|
| 156 |
|
| 157 |
+
return short_output, long_output, fig
|
| 158 |
|
| 159 |
|
| 160 |
if __name__ == "__main__":
|