Update predict_model.py

src/models/predict_model.py

@@ -1,6 +1,7 @@
 import yaml
 
 from .model import Summarization
+import pandas as pd
 
 def predict_model(text):
     """