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