Update app.py

app.py

@@ -41,9 +41,10 @@ if st.button("Predict"):
     trans = vectorizer.transform([text])
     body_len = len(text) - text.count(" ")
     punct = count_punct(text)
-    k = {"body_len": [body_len], "punc%": [punct]}
+    #k = {"body_len": [body_len], "punc%": [punct]}
+    k = {body_len: [body_len], punc%: [punct]}
     df = pd.DataFrame(k)
-    df.columns = df.columns.astype(str) 
+    #df.columns = df.columns.astype(str) 
     test_vect = pd.concat([df.reset_index(drop=True),
                            pd.DataFrame(trans.toarray())], axis=1)
     prediction = model.predict(test_vect)