Update my_model/tabs/results.py

my_model/tabs/results.py

@@ -64,7 +64,7 @@ class ResultDemonstrator(KBVQAEvaluator):
 
         plt.figure(figsize=(10, 6))
         # Create a scatter plot with smaller dots using the 's' parameter
-        scatter = plt.scatter(range(len(token_counts)), token_counts, c=colors, s=20, label=labels)
+        scatter = plt.scatter(range(len(token_counts)), token_counts, c=d['color'], s=20, label=d['label'])
         
         # Create a custom legend
         from matplotlib.lines import Line2D