Upload models.py

models.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import pickle
+  
 
 # Load the trained model
 with open('model_xgb.pkl', 'rb') as file_1:
@@ -50,7 +51,7 @@ def run():
 
     # Make prediction
     if submitted:
-        prediction = model.predict(df)
+        prediction = model_inf_xgb.predict(df)
 
         # Display prediction result
         if prediction[0] == 0: