Update app.py

app.py

@@ -64,7 +64,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]] = math.ceil(max_scores * 1000) / 1000.0
+        df_cosines[df_category.iloc[0,2]] = np.round(max_scores * 1000) / 1000.0
         for num_sentence, scores in enumerate(cosine_scores):
             for num_phrase, score in enumerate(scores):
                 if score >= threshold:
@@ -73,7 +73,7 @@ def compare_text(transcript, categories, threshold):
                         'sentence': sentences[num_sentence],
                         'phrase': df_category.at[num_phrase,'example'],
                         'category': df_category.at[num_phrase,'label'],
-                        'similarity': math.ceil(score * 1000) / 1000.0
+                        'similarity': np.round(score * 1000) / 1000.0
                     }
                     df_results = df_results.append(new_row, ignore_index=True)