Update app.py

app.py

@@ -12,6 +12,7 @@ from typing import Dict, List, Optional, Tuple
 from dataclasses import dataclass
 from pathlib import Path
 import logging
+from plotly.graph_objects import Figure
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -214,7 +215,7 @@ class AfriMMLUEvaluator:
 
 class VisualizationManager:
     @staticmethod
-    def create_visualization(results: EvaluationResult) -> Tuple[pd.DataFrame, px.Figure]:
+    def create_visualization(results: EvaluationResult) -> Tuple[pd.DataFrame, Figure]:
         """Create visualization from evaluation results."""
         summary_data = [
             {'Subject': subject, 'Accuracy (%)': accuracy}