Update predict_model.py

src/models/predict_model.py

@@ -19,6 +19,6 @@ def predict_model(text):
 
 
 if __name__ == '__main__':
-    text = make_dataset(split="test")['input_text'][0]
+    text = pd.load_csv('data/processed/test.csv')['input_text'][0]
     pre_summary = predict_model(text)
     print(pre_summary)