Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import pandas as pd
-import pickle
+import joblib
 
 st.header('FTDS Model Deployment')
 st.write("""
@@ -45,8 +45,7 @@ input = user_input()
 st.subheader('User Input')
 st.write(input)
 
-with open("my_model.pkl", "rb") as f:
-    load_model = pickle.load(f)
+load_model = joblib.load("my_model.pkl")
 
 if st.button("Predict"):
     prediction = load_model.predict(input)