Spaces:
Runtime error
Runtime error
File size: 477 Bytes
b5ad754 092c7c9 b5ad754 092c7c9 6f26bdf 291e845 6f26bdf b5ad754 092c7c9 080261d 6f26bdf b5ad754 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from rouge_score import rouge_scorer
def calc_rouge_score(summary, reference):
scorer = rouge_scorer.RougeScorer(['rouge1', 'rouge2', 'rougeL', 'rougeLsum'], use_stemmer=True)
scores = scorer.score(summary, reference)
text = f""
for s in scores:
text += f"The {s}-score is {scores[s].fmeasure}\n"
return text
iface = gr.Interface(
fn=calc_rouge_score,
inputs=["text", "text"],
outputs=["text"])
iface.launch()
|