dprat0821's picture
Update app.py
7b08134 verified
raw
history blame contribute delete
904 Bytes
import gradio as gr
import pandas as pd
from evaluate.visualization import radar_plot # 正确模块
# 模拟的评分数据
scores = {
"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))
fig.suptitle("GRACE 模型评估对比图", fontsize=14)
return fig
# Gradio 界面构建
with gr.Blocks() as demo:
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()