Update app.py

app.py

@@ -175,13 +175,14 @@ class CombinedProcessor:
         bibliography_entries = editorial_df[editorial_df['entity_group'] == 'bibliography']['word'].tolist()
         
         bibtex_entries = []
-
+        list_style = []
+        
         for entry in bibliography_entries:
             print(entry)
             entry = re.sub(r'- ?[\n¶] ?', r'', entry)
             entry = re.sub(r' ?[\n¶] ?', r' ', entry)
-            style = style_classifier(entry, truncation=True, padding=True, top_k=10)
-            print(style)
+            style = pd.DataFrame(style_classifier(entry, truncation=True, padding=True, top_k=1))
+            list_style.append(style)
             entry = re.sub(r'\s*([;:,\.])\s*', r' \1 ', entry)
             #print(entry)
             bib_out = bibliography_classifier(entry)
@@ -208,6 +209,8 @@ class CombinedProcessor:
             bibtex_entry = create_bibtex_entry(bibtex_data)
             bibtex_entries.append(bibtex_entry)
 
+        list_style = pd.concat(list_style)
+        print(list_style)
         #Disambiguation to avoid duplicate ids.
         bibtex_entries = disambiguate_bibtex_ids(bibtex_entries)