MarMont commited on
Commit
95dd02a
1 Parent(s): c8acb18

debug statements

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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