Update my_model/tabs/results.py

my_model/tabs/results.py

@@ -6,6 +6,7 @@ from ast import literal_eval
 from typing import Union, List
 import streamlit as st
 from my_model.results.evaluation import KBVQAEvaluator
+from my_model.config import evaluation_config as config 
 
 class ResultDemonstrator(KBVQAEvaluator):
 
@@ -70,10 +71,12 @@ class ResultDemonstrator(KBVQAEvaluator):
 
 
 ####################
+        scores = d['vqa_score_13b_caption+detic']
+        token_counts = d['trimmed_tokens_count_caption_detic']
 
         # Define colors and labels for the legend
-        colors = ['green' if accuracy == 1 else 'orange' if round(accuracy,2) == 0.67 else 'red' for accuracy in accuracies]
-        labels = ['Correct' if accuracy == 1 else 'Partially Correct' if accuracy == 0.67 else 'Incorrect' for accuracy in accuracies]
+        colors = ['green' if score == 1 else 'orange' if round(score,2) == 0.67 else 'red' for score in scores]
+        labels = ['Correct' if score == 1 else 'Partially Correct' if score == 0.67 else 'Incorrect' for score in scores]
         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)