Update app.py

app.py

@@ -236,7 +236,7 @@ if query:
                 # text=f"Top 10000 words in an interactive embedding map for {query} in {database_name} PubMed corpus"
                 #      f": Zoom in to the black diamond to find {query}", x=0.5, y=1, xanchor='center', yanchor='top',
                 # font=dict(color='black')))
-            fig.update_coloraxes(colorbar_title="Similarity with query")
+            fig.update_coloraxes(colorbar_title=f"Similarity with {query}")
 
             # Represent query as a large red diamond
             fig.add_trace(go.Scatter(x=[model.wv[query][0]], y=[model.wv[query][1]], mode='markers',
@@ -508,7 +508,7 @@ if query:
                 title=dict(text=f"", x=0.5, y=0.95,
                            xanchor='center', yanchor='top', font=dict(color='black')),
                 scene=dict(xaxis_title="Dimension 1", yaxis_title="Dimension 2", zaxis_title="Dimension 3"))
-        fig2.update_coloraxes(colorbar_title="Similarity with query")
+        fig2.update_coloraxes(colorbar_title=f"Similarity with {query}")
 
         # Represent query as a large red diamond
         fig2.add_trace(