Update app.py

app.py

@@ -90,6 +90,9 @@ def process_classification(text: str, model1, model2, tokenizer1) -> Tuple[str,
 
     return f"{round(prediction1, 1)}", f"{round(prediction2, 1)}", f"{round(score, 2)}"
 
+import plotly.graph_objects as go
+from typing import Tuple
+
 def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figure, go.Figure]:
     entities_bin = [entity['entity'] for entity in ner_output_bin['entities']]
     entities_ext = [entity['entity'] for entity in ner_output_ext['entities']]
@@ -104,7 +107,6 @@ def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figu
     ext_labels = list(entity_counts_ext.keys())
     ext_sizes = list(entity_counts_ext.values())
 
-
     # Define color mapping
     bin_color_map = {
         "External": "#6ad5bc",
@@ -127,7 +129,12 @@ def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figu
     
     # Create pie chart for extended classification
     fig1 = go.Figure(data=[go.Pie(labels=ext_labels, values=ext_sizes, textinfo='label+percent', hole=.3, marker=dict(colors=ext_colors))])
-    fig1.update_layout(title_text='Extended Sequence Classification Subclasses', template='plotly_dark')
+    fig1.update_layout(
+        title_text='Extended Sequence Classification Subclasses',
+        template='plotly_dark',
+        plot_bgcolor='rgba(0,0,0,0)',
+        paper_bgcolor='rgba(0,0,0,0)'
+    )
     
     # Create bar chart for binary classification
     fig2 = go.Figure(data=[go.Bar(x=bin_labels, y=bin_sizes, marker=dict(color=bin_colors))])
@@ -135,7 +142,9 @@ def generate_charts(ner_output_bin: dict, ner_output_ext: dict) -> Tuple[go.Figu
         title='Binary Sequence Classification Classes',
         xaxis_title='Entity Type',
         yaxis_title='Count',
-        template='plotly_dark'
+        template='plotly_dark',
+        plot_bgcolor='rgba(0,0,0,0)',
+        paper_bgcolor='rgba(0,0,0,0)'
     )
     
     return fig1, fig2