Hailing / app.py
hl250's picture
Add application
d956513 verified
Raw
History Blame Contribute Delete
1.27 kB
import gradio as gr
from transformers import pipeline
# 加载作文评分模型(替换为你的模型名)
chef = pipeline("text-classification", model="hl250/roberta-base")
def score_essay(essay):
"""评分函数"""
try:
result = chef(essay)
score = result[0]['score']
# 根据分数给出评价
if score >= 20:
feedback = "🎉 优秀!内容充实,表达清晰。"
elif score >= 15:
feedback = "👍 良好。可以进一步丰富内容。"
elif score >= 10:
feedback = "💪 及格。需要改进结构和表达。"
else:
feedback = "📝 需要大幅提升写作水平。"
return f"📝 作文得分: {score:.1f}/30\n\n💡 评语: {feedback}"
except Exception as e:
return f"❌ 评分出错: {str(e)}"
# 创建界面
demo = gr.Interface(
fn=score_essay,
inputs=gr.Textbox(
lines=10,
placeholder="请输入作文内容...",
label="作文内容"
),
outputs=gr.Textbox(label="评分结果"),
title="🤖 AI作文评分系统",
description="输入作文内容,获取AI自动评分和评语"
)
# 启动应用
demo.launch()