pentarosarium commited on
Commit
268a755
·
1 Parent(s): 9fd374e

progress more 28

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -142,14 +142,14 @@ def process_file(uploaded_file):
142
  progress_text.text(f"{i + 1} из {total_news} сообщений предобработано")
143
 
144
  # Perform sentiment analysis
145
- rubert1_results = [get_rubert1_sentiment(text) for text in texts]
146
  rubert2_results = [get_rubert2_sentiment(text) for text in texts]
147
  finbert_results = [get_finbert_sentiment(text) for text in translated_texts]
148
  roberta_results = [get_roberta_sentiment(text) for text in translated_texts]
149
  finbert_tone_results = [get_finbert_tone_sentiment(text) for text in translated_texts]
150
 
151
  # Add results to DataFrame
152
- df['ruBERT1'] = rubert1_results
153
  df['ruBERT2'] = rubert2_results
154
  df['FinBERT'] = finbert_results
155
  df['RoBERTa'] = roberta_results
@@ -157,13 +157,13 @@ def process_file(uploaded_file):
157
  df['Translated'] = translated_texts
158
 
159
  # Reorder columns
160
- columns_order = ['Объект', 'ruBERT1', 'ruBERT2','FinBERT', 'RoBERTa', 'FinBERT-Tone', 'Выдержки из текста', 'Translated' ]
161
  df = df[columns_order]
162
 
163
  return df
164
 
165
  def main():
166
- st.title("... приступим к анализу... версия 27")
167
 
168
  uploaded_file = st.file_uploader("Выбирайте Excel-файл", type="xlsx")
169
 
 
142
  progress_text.text(f"{i + 1} из {total_news} сообщений предобработано")
143
 
144
  # Perform sentiment analysis
145
+ #rubert1_results = [get_rubert1_sentiment(text) for text in texts]
146
  rubert2_results = [get_rubert2_sentiment(text) for text in texts]
147
  finbert_results = [get_finbert_sentiment(text) for text in translated_texts]
148
  roberta_results = [get_roberta_sentiment(text) for text in translated_texts]
149
  finbert_tone_results = [get_finbert_tone_sentiment(text) for text in translated_texts]
150
 
151
  # Add results to DataFrame
152
+ #df['ruBERT1'] = rubert1_results
153
  df['ruBERT2'] = rubert2_results
154
  df['FinBERT'] = finbert_results
155
  df['RoBERTa'] = roberta_results
 
157
  df['Translated'] = translated_texts
158
 
159
  # Reorder columns
160
+ columns_order = ['Объект', 'ruBERT2','FinBERT', 'RoBERTa', 'FinBERT-Tone', 'Выдержки из текста', 'Translated' ]
161
  df = df[columns_order]
162
 
163
  return df
164
 
165
  def main():
166
+ st.title("... приступим к анализу... версия 28")
167
 
168
  uploaded_file = st.file_uploader("Выбирайте Excel-файл", type="xlsx")
169