debug statements
Browse files
app.py
CHANGED
|
@@ -511,6 +511,7 @@ def main(dataset, model, progress=gr.Progress(track_tqdm=True)):
|
|
| 511 |
keywords = '(' + ' OR '.join(keyword_list) + ')'
|
| 512 |
else:
|
| 513 |
keywords = keyword_list[0]
|
|
|
|
| 514 |
if dataset in examples:
|
| 515 |
df = get_example(keywords)
|
| 516 |
place_data = 'test'
|
|
@@ -520,12 +521,15 @@ def main(dataset, model, progress=gr.Progress(track_tqdm=True)):
|
|
| 520 |
print(df)
|
| 521 |
|
| 522 |
if model == 'LDA':
|
|
|
|
| 523 |
top_tweets = full_lda(df)
|
|
|
|
| 524 |
place_data = 'test'
|
| 525 |
else:
|
| 526 |
base_bertopic()
|
| 527 |
optimized_bertopic()
|
| 528 |
|
|
|
|
| 529 |
headlines = topic_summarization(top_tweets)
|
| 530 |
headlines = '\n'.join(str(h) for h in headlines)
|
| 531 |
|
|
|
|
| 511 |
keywords = '(' + ' OR '.join(keyword_list) + ')'
|
| 512 |
else:
|
| 513 |
keywords = keyword_list[0]
|
| 514 |
+
|
| 515 |
if dataset in examples:
|
| 516 |
df = get_example(keywords)
|
| 517 |
place_data = 'test'
|
|
|
|
| 521 |
print(df)
|
| 522 |
|
| 523 |
if model == 'LDA':
|
| 524 |
+
print('doing lda')
|
| 525 |
top_tweets = full_lda(df)
|
| 526 |
+
print('done lda')
|
| 527 |
place_data = 'test'
|
| 528 |
else:
|
| 529 |
base_bertopic()
|
| 530 |
optimized_bertopic()
|
| 531 |
|
| 532 |
+
print('doing topic summarization')
|
| 533 |
headlines = topic_summarization(top_tweets)
|
| 534 |
headlines = '\n'.join(str(h) for h in headlines)
|
| 535 |
|