Update classification.py

classification.py

@@ -172,17 +172,16 @@ def process_categories(categories, model):
 
 def match_categories(df, category_df, treshold=0.45):
     for topic in category_df['topic']:
-		df[topic] = 0
+        df[topic] = 0
     for i, ebd_content in enumerate(df['Embeddings']):
         if isinstance(ebd_content, torch.Tensor):
             cos_scores = util.cos_sim(ebd_content, torch.stack(list(category_df['Embeddings']), dim=0))[0]
             high_score_indices = [i for i, score in enumerate(cos_scores) if score > treshold]
-			for j in high_score_indices:
-				df.loc[i, category_df.loc[j, 'topic']] = float(cos_scores[index])
+            for j in high_score_indices:
+                df.loc[i, category_df.loc[j, 'topic']] = float(cos_scores[index])
     return df
 
 def save_data(df, filename):
-
     df = df.drop(columns=['Embeddings'])
     new_filename = filename.replace(".", "_classified.")
     df.to_excel(new_filename, index=False)