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 |
|