Update app.py

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
 import numpy as np
 import pandas as pd
-#import joblib
+import joblib
 import pickle
 
-model = pickle.load(open("PycaretGBR.pkl", 'rb'))
+# model = pickle.load(open("PycaretGBR.pkl", 'rb'))
+model = joblib.load("lr.joblib")
 st.title('Developer Salary Prediction 2024')
 st.write("""### We need some information to predict the salary""")
 
@@ -59,8 +60,7 @@ columns = ['Country', 'EdLevel', 'YearsCodePro', 'Employment']
 
 ok = st.button("Calculate Salary")
 if ok:
-    #X_new = np.array([countries,education,expericence,employment])
-    X_new_df = pd.DataFrame([[countries,education,expericence,employment]],
+    X_new_df = pd.DataFrame([[country,education,expericence,employment]],
                             columns = ['Country', 'EdLevel', 'YearsCodePro', 'Employment'])
     print("##########")
     print("##########")