Update app.py

app.py

@@ -209,7 +209,7 @@ def predict(text_input, progress=gr.Progress()):
 
     plot = datamapplot.create_interactive_plot(
         stacked_df[['x','y']].values,
-        np.array(basedata_df['cluster_1_labels']),
+        np.array(stacked_df['cluster_1_labels']),np.array(stacked_df['cluster_2_labels']),np.array(stacked_df['cluster_3_labels']),
         hover_text=[str(ix) + ', ' + str(row['parsed_publication']) + str(row['title']) for ix, row in stacked_df.iterrows()],
         font_family="Roboto Condensed",marker_color_array=stacked_df['color']
     )