sergiomar73 commited on
Commit
9ab545f
1 Parent(s): 8ed8ae5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,8 +57,8 @@ def compare_text(transcript, categories):
57
  # Compute cosine-similarities
58
  cosine_scores = util.cos_sim(embeddings, phrases).numpy()
59
  max_scores = np.max(cosine_scores, axis=1)
60
- df_results_plot[df_category.iloc[0,2]] = math.ceil(max_scores * 100) / 100.0 # .round(decimals = 2)
61
- df_results_grid[df_category.iloc[0,2]] = math.ceil(max_scores * 1000) / 1000.0 # .round(decimals = 3)
62
  # Plot
63
  fig = px.imshow(
64
  df_results_plot,
 
57
  # Compute cosine-similarities
58
  cosine_scores = util.cos_sim(embeddings, phrases).numpy()
59
  max_scores = np.max(cosine_scores, axis=1)
60
+ df_results_plot[df_category.iloc[0,2]] = np.round(max_scores * 100) / 100.0 # .round(decimals = 2)
61
+ df_results_grid[df_category.iloc[0,2]] = np.round(max_scores * 1000) / 1000.0 # .round(decimals = 3)
62
  # Plot
63
  fig = px.imshow(
64
  df_results_plot,