gagan3012 commited on
Commit
c9f9b78
·
1 Parent(s): 8d6cb30

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +0 -1
src/models/predict_model.py CHANGED
@@ -15,6 +15,5 @@ def predict_model(text: str):
15
  pre_summary = model.predict(text)
16
  return pre_summary
17
 
18
- text = pd.load_csv("data/processed/test.csv")["input_text"][0]
19
  pre_summary = predict_model(text)
20
  print(pre_summary)
 
15
  pre_summary = model.predict(text)
16
  return pre_summary
17
 
 
18
  pre_summary = predict_model(text)
19
  print(pre_summary)