Gagan Bhatia commited on
Commit
f68a895
1 Parent(s): 5ffa1ac

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -458,3 +458,4 @@ class Summarization:
458
  metrics: str = "rouge"
459
  ):
460
  metric = load_metric(metrics)
 
 
458
  metrics: str = "rouge"
459
  ):
460
  metric = load_metric(metrics)
461
+ input_text = test_df['input_text']