Update my_model/tabs/results.py

my_model/tabs/results.py

@@ -59,23 +59,8 @@ class ResultDemonstrator(KBVQAEvaluator):
         # Compute colors and labels for the plot (assuming these columns are already in the DataFrame)
         d['color'] = d['vqa_score_13b_caption+detic'].apply(lambda x: 'green' if x == 1 else 'orange' if round(x, 2) == 0.67 else 'red')
         d['label'] = d['vqa_score_13b_caption+detic'].apply(lambda x: 'Correct' if x == 1 else 'Partially Correct' if x == 0.67 else 'Incorrect')
-        
-        # Creating the scatter plot
-        scatter_chart = alt.Chart(d).mark_circle(size=20).encode(
-            x=alt.X('index:Q', title='Index'),  # Assuming 'index' is a column or can be created
-            y=alt.Y('token_counts:Q', title='Number of Tokens'),
-            color=alt.Color('color:N', legend=alt.Legend(title="VQA Score")),
-            tooltip=['index', 'token_counts', 'label']
-        ).interactive()
-        
-        # Display the chart
-        st.altair_chart(scatter_chart, use_container_width=True)
-
-
-####################
-        scores = d['vqa_score_13b_caption+detic']
 
-        # Define colors and labels for the legend
+                # Define colors and labels for the legend
 
         plt.figure(figsize=(10, 6))
         # Create a scatter plot with smaller dots using the 's' parameter
@@ -94,4 +79,25 @@ class ResultDemonstrator(KBVQAEvaluator):
         plt.ylabel('Number of Tokens')
         
         # Display the plot
-        plt.show()
+        plt.show()
+
+
+
+
+
+
+        
+        # Creating the scatter plot
+        scatter_chart = alt.Chart(d).mark_circle(size=20).encode(
+            x=alt.X('index:Q', title='Index'),  # Assuming 'index' is a column or can be created
+            y=alt.Y('token_counts:Q', title='Number of Tokens'),
+            color=alt.Color('color:N', legend=alt.Legend(title="VQA Score")),
+            tooltip=['index', 'token_counts', 'label']
+        ).interactive()
+        
+        # Display the chart
+        st.altair_chart(scatter_chart, use_container_width=True)
+
+
+####################
+