Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from sentence_transformers import SentenceTransformer, util
 import numpy as np
 import pandas as pd
+import math
 import spacy
 import torch
 import plotly.express as px
@@ -56,10 +57,8 @@ def compare_text(transcript, categories):
         # Compute cosine-similarities
         cosine_scores = util.cos_sim(embeddings, phrases).numpy()
         max_scores = np.max(cosine_scores, axis=1)
-        df_results_plot[df_category.iloc[0,2]] = max_scores
-        df_results_grid[df_category.iloc[0,2]] = max_scores
-    df_results_plot = df_results_plot.round(decimals = 2)
-    df_results_grid = df_results_grid.round(decimals = 3) 
+        df_results_plot[df_category.iloc[0,2]] = math.ceil(max_scores * 100) / 100.0   # .round(decimals = 2)
+        df_results_grid[df_category.iloc[0,2]] = math.ceil(max_scores * 1000) / 1000.0 # .round(decimals = 3)
     # Plot
     fig = px.imshow(
         df_results_plot,