Update predict_model.py

src/models/predict_model.py

@@ -15,7 +15,6 @@ def predict_model(text: str):
     pre_summary = model.predict(text)
     return pre_summary
 
-if __name__ == "__main__":
     text = pd.load_csv("data/processed/test.csv")["input_text"][0]
     pre_summary = predict_model(text)
     print(pre_summary)