Update app.py

app.py

@@ -63,7 +63,7 @@ def compare_text(transcript, categories, threshold):
         phrases = torch.stack(phrases_list)
         cosine_scores = util.cos_sim(embeddings, phrases).numpy()
         max_scores = np.max(cosine_scores, axis=1)
-        df_cosines[df_category.iloc[0,2]] = max_scores
+        df_cosines[df_category.iloc[0,2]] = max_scores.round(decimals = 3)
         for num_sentence, scores in enumerate(cosine_scores):
             for num_phrase, score in enumerate(scores):
                 if score >= threshold:
@@ -76,7 +76,7 @@ def compare_text(transcript, categories, threshold):
                     }
                     df_results = df_results.append(new_row, ignore_index=True)
 
-    df_cosines = df_cosines.round(decimals = 3) 
+    # df_cosines = df_cosines.round(decimals = 3) 
     df_results = df_results.sort_values(['line','similarity'],ascending=[True,False]).round(decimals = 3)
 
     df_summary = pd.DataFrame(df_cosines.max(numeric_only=True),columns=['similarity'])