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