kapllan commited on
Commit
604d17a
1 Parent(s): 570d236

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -137,10 +137,12 @@ def restructure_topics(topics: List[Tuple[str, float]]) -> List[Tuple[str, str,
137
  for predicted_main_topic, predicted_sub_topics in topics_as_dict.items():
138
  if len(predicted_sub_topics) > 0:
139
  score = [t for t in topics if t[0] == predicted_main_topic][0][1]
 
140
  topics_restructured.append(("Main Topic", predicted_main_topic, score))
141
  predicted_sub_topics_with_scores = []
142
  for pst in predicted_sub_topics:
143
  score = [t for t in topics if t[0] == pst][0][1]
 
144
  entry = ("Sub Topic", pst, score)
145
  if entry not in predicted_sub_topics_with_scores:
146
  predicted_sub_topics_with_scores.append(entry)
 
137
  for predicted_main_topic, predicted_sub_topics in topics_as_dict.items():
138
  if len(predicted_sub_topics) > 0:
139
  score = [t for t in topics if t[0] == predicted_main_topic][0][1]
140
+ predicted_main_topic = predicted_main_topic.replace("hauptthema: ", "")
141
  topics_restructured.append(("Main Topic", predicted_main_topic, score))
142
  predicted_sub_topics_with_scores = []
143
  for pst in predicted_sub_topics:
144
  score = [t for t in topics if t[0] == pst][0][1]
145
+ pst = pst.replace("unterthema: ", "")
146
  entry = ("Sub Topic", pst, score)
147
  if entry not in predicted_sub_topics_with_scores:
148
  predicted_sub_topics_with_scores.append(entry)