Update app.py

app.py

@@ -1,22 +1,29 @@
 import gradio as gr
-import numpy as np
 import pandas as pd
-from evaluate import load
-from evaluate.visualization import radar_plot
+from evaluate.visualization import radar_plot  # 正确模块
 
-# 假设这是两个模型在 GRACE 维度上的评分
+# 模拟的评分数据
 scores = {
-    "Model A": [4.0, 4.5, 3.5, 4.0],  # e.g., G, R, A, E
+    "Model A": [4.0, 4.5, 3.5, 4.0],
     "Model B": [3.5, 4.0, 4.0, 3.5]
 }
 labels = ["Generalization", "Relevance", "Artistry", "Efficiency"]
 df = pd.DataFrame(scores, index=labels).T
 
+# 图表绘制函数
+def plot_radar():
+    data = []
+    for model in df.index:
+        data.append({label: df.loc[model, label] for label in df.columns})
+    fig = radar_plot(data=data, model_names=list(df.index), title="GRACE 评估维度雷达图")
+    return fig
+
+# Gradio 界面构建
 with gr.Blocks() as demo:
-    gr.Markdown("## 模型对比雷达图")
-    plot = gr.Plot()
-    def plot_radar():
-        fig = radar_plot(df, title="GRACE 维度对比")
-        return fig
-    plot.load(plot_radar)
+    gr.Markdown("## ✨ 模型 GRACE 维度雷达图")
+    with gr.Row():
+        generate_button = gr.Button("生成图表")
+    output_plot = gr.Plot()
+    generate_button.click(fn=plot_radar, inputs=[], outputs=output_plot)
+
 demo.launch()