Commit
·
9ab545f
1
Parent(s):
8ed8ae5
Update app.py
Browse files
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]] =
|
61 |
-
df_results_grid[df_category.iloc[0,2]] =
|
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,
|