Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Jul 27, 2021

Commit

f9d6102

1 Parent(s): 7865abb

fixes

Files changed (1) hide show

src/models/evaluate_model.py CHANGED Viewed

@@ -13,7 +13,7 @@ def evaluate_model():
     with open("params.yml") as f:
         params = yaml.safe_load(f)
-    test_df = pd.read_csv('data/processed/test.csv')
     model = Summarization()
     model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
     results = model.evaluate(test_df=test_df, metrics=params['metric'])

     with open("params.yml") as f:
         params = yaml.safe_load(f)
+    test_df = pd.read_csv('data/processed/test.csv')[:25]
     model = Summarization()
     model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
     results = model.evaluate(test_df=test_df, metrics=params['metric'])