Upload folder using huggingface_hub

app.py

@@ -58,9 +58,9 @@ data = pd.DataFrame({
 })
 
 if st.button("Predict"):
-  prediction = model.predict(data)[0,1]
+  prediction = model.predict(data)[0]
   pred =  "purchase" if (prediction == 1 ) else "not purchase"
-  st.write("Based on the prediction, the customer will", pred,"the travel product")
+  st.write("Based on the prediction, the customer will ", pred," the travel product")