Update app.py

app.py

@@ -122,13 +122,16 @@ def pre_process(df):
   df['opo_pre_tkn'] = pre_processed_data
   return df
 
-def classify(df, new_column = True):
+def classify(df, new_column = True, pre_processed = False):
   sentencesMCTIList_xp8 = df['opo_pre_tkn']
-
+    
   formatted_sentences = []
-  for sentence in sentencesMCTIList_xp8:
-    formatted_sentences.append(json.loads(sentence.replace("'",'"')))
-  del sentencesMCTIList_xp8
+  if not pre_processed:
+    for sentence in sentencesMCTIList_xp8:
+      formatted_sentences.append(json.loads(sentence.replace("'",'"')))
+    del sentencesMCTIList_xp8
+  else:
+    formatted_sentences = sentencesMCTIList_xp8
 
   words = list(reloaded_w2v_model.wv.vocab) 
   item_shape = np.shape(reloaded_w2v_model.wv[words[0]])
@@ -180,7 +183,7 @@ def app(operacao, resultado, dados):
 
   if operacao == "Pré-processamento + Classificação" :
     preprocessed = pre_process(data)
-    df = classify(preprocessed, resultado == "Nova Coluna")
+    df = classify(preprocessed, resultado == "Nova Coluna", pre_processed = True)
     output = gen_output(df)
     
     return output