Update app.py

app.py

@@ -1,11 +1,13 @@
 import streamlit as st
+import pandas as pd
 import pickle
 
 with open('./trained_model.pkl', 'rb') as file:
     loaded_model = pickle.load(file)
 
-input = st.text_area('Enter text')
+new_pred = st.text_area('Enter text')
 
-if input:
-    out = loaded_model(input)
+if new_pred:
+    new_pred_processed = generate_features(pd.DataFrame([new_pred]), prod=True)
+    out = loaded_model(new_pred_processed)
     st.json(out[0][0])