Update app.py

app.py

@@ -248,7 +248,12 @@ with similarity_tab:
     
     if text_input and selected_model:
         with st.spinner("Computing token similarities..."):
+            # Ensure unique token names by appending their index
+            unique_tokens = [f"{token}_{i}" for i, token in enumerate(tokens)]
+            
             sim_df = compute_token_similarities(embeddings, tokens)
+            sim_df.columns = unique_tokens  # Update column names
+            sim_df.index = unique_tokens  # Update row names
             
             fig = px.imshow(
                 sim_df,
@@ -277,7 +282,7 @@ with similarity_tab:
             
             for token1, token2, sim in top_pairs:
                 st.write(f"'{token1}' — '{token2}': {sim:.3f}")
-
+                
 st.markdown("---")
 st.markdown("""
     💡 **Tips:**