Pclanglais commited on
Commit
86c4d11
·
verified ·
1 Parent(s): 8e58d65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -168,8 +168,6 @@ class CombinedProcessor:
168
 
169
  editorial_out = editorial_classifier(batch_prompts)
170
  editorial_df = pd.concat([pd.DataFrame(classification) for classification in editorial_out])
171
-
172
- print(editorial_df)
173
 
174
  # Filter out only bibliography entries
175
  bibliography_entries = editorial_df[editorial_df['entity_group'] == 'bibliography']['word'].tolist()
@@ -210,7 +208,10 @@ class CombinedProcessor:
210
  bibtex_entries.append(bibtex_entry)
211
 
212
  list_style = pd.concat(list_style)
213
- print(list_style)
 
 
 
214
  #Disambiguation to avoid duplicate ids.
215
  bibtex_entries = disambiguate_bibtex_ids(bibtex_entries)
216
 
 
168
 
169
  editorial_out = editorial_classifier(batch_prompts)
170
  editorial_df = pd.concat([pd.DataFrame(classification) for classification in editorial_out])
 
 
171
 
172
  # Filter out only bibliography entries
173
  bibliography_entries = editorial_df[editorial_df['entity_group'] == 'bibliography']['word'].tolist()
 
208
  bibtex_entries.append(bibtex_entry)
209
 
210
  list_style = pd.concat(list_style)
211
+ list_style = list_style.groupby('label')['score'].mean().sort_values(ascending=False).reset_index()
212
+ top_style = result.iloc[0]['label']
213
+ print(top_style)
214
+
215
  #Disambiguation to avoid duplicate ids.
216
  bibtex_entries = disambiguate_bibtex_ids(bibtex_entries)
217