Update app.py

app.py

@@ -236,10 +236,10 @@ if query:
                               yaxis=dict(gridcolor='#CCFFFF', color='blue'))
 
             # fig.update_traces(hovertemplate='<b>%{hovertext}</b><br>Similarity score: %{customdata[0]:.2f}<extra></extra>')
-            # fig.update_layout(title=dict(
-            #     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=.8, xanchor='center', yanchor='top',
-            #     font=dict(color='black')))
+            fig.update_layout(title=dict(
+                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=.8, xanchor='center', yanchor='top',
+                font=dict(color='black')))
             fig.update_coloraxes(colorbar_title="Similarity with query")
 
             # Represent query as a large red diamond
@@ -506,8 +506,8 @@ if query:
         fig2.update_traces(
                 hovertemplate='<b>%{hovertext}</b><br>Similarity score: %{customdata[0]:.2f}<extra></extra>')
         fig2.update_layout(
-                # title=dict(text=f"Word embedding map for {query} in {database_name} PubMed corpus", x=0.5, y=0.8,
-                #            xanchor='center', yanchor='top', font=dict(color='black')),
+                title=dict(text=f"Word embedding map for {query} in {database_name} PubMed corpus", x=0.5, y=0.8,
+                           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")