Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import pandas as pd
 import json
 import os
-from evo_transformer import EvoTransformer
 from plot import plot_radar_chart
-from diagram import plot_architecture_diagram
+from diagram import show_transformer_diagram
+from evo_transformer import EvoTransformer
 
 # === Initialize ===
 evo = EvoTransformer()