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()