Gagan Bhatia commited on
Commit
7d8eeca
1 Parent(s): a739a9d

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +1 -0
src/models/predict_model.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from .model import Summarization
2
 
3
  def predict_model(text):
 
1
+ from src.data.make_dataset import make_dataset
2
  from .model import Summarization
3
 
4
  def predict_model(text):