Spaces:
Runtime error
Runtime error
fixes
Browse files- src/models/model.py +2 -3
src/models/model.py
CHANGED
@@ -482,7 +482,7 @@ class Summarization:
|
|
482 |
print(type(predictions), type(references))
|
483 |
|
484 |
results = metric.compute(predictions=predictions, references=references)
|
485 |
-
|
486 |
output = {
|
487 |
'Rouge 1': {
|
488 |
'Rouge_1 Low Precision': results["rouge1"].low.precision,
|
@@ -529,5 +529,4 @@ class Summarization:
|
|
529 |
'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
|
530 |
}
|
531 |
}
|
532 |
-
|
533 |
-
return results
|
|
|
482 |
print(type(predictions), type(references))
|
483 |
|
484 |
results = metric.compute(predictions=predictions, references=references)
|
485 |
+
|
486 |
output = {
|
487 |
'Rouge 1': {
|
488 |
'Rouge_1 Low Precision': results["rouge1"].low.precision,
|
|
|
529 |
'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
|
530 |
}
|
531 |
}
|
532 |
+
return output
|
|