fix: Adapt labels creation with interactive plot implementation

app.py

@@ -41,8 +41,6 @@ def draw_interactive_scatter_plot(
     texts: np.ndarray, xs: np.ndarray, ys: np.ndarray, values: np.ndarray, labels: np.ndarray, text_column: str, label_column: str
 ) -> Figure:
     # Normalize values to range between 0-255, to assign a color for each value
-    values = np.array(values)
-    labels = np.array(labels)
     max_value = values.max()
     min_value = values.min()
     if max_value - min_value == 0:
@@ -80,7 +78,7 @@ def generate_plot(
     with st.spinner("Reducing dimensionality..."):
         embeddings_2d = get_tsne_embeddings(embeddings)
     plot = draw_interactive_scatter_plot(
-        df, embeddings_2d[:, 0], embeddings_2d[:, 1], encoded_labels, labels, 'text', 'label'
+        df, embeddings_2d[:, 0], embeddings_2d[:, 1], encoded_labels, encoded_labels, 'text', 'label'
     )
     return plot