Gagan Bhatia commited on
Commit
56499fc
1 Parent(s): bd01d65

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +2 -2
src/models/predict_model.py CHANGED
@@ -17,7 +17,7 @@ def predict_model(text):
17
  return pre_summary
18
 
19
 
20
- if __name__ == '__main__':
21
- text = pd.load_csv('data/processed/test.csv')['input_text'][0]
22
  pre_summary = predict_model(text)
23
  print(pre_summary)
 
17
  return pre_summary
18
 
19
 
20
+ if __name__ == "__main__":
21
+ text = pd.load_csv("data/processed/test.csv")["input_text"][0]
22
  pre_summary = predict_model(text)
23
  print(pre_summary)