hackerbyhobby commited on
Commit
28d8ff1
·
unverified ·
1 Parent(s): 32e56b3

forgot to init

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import joblib
3
  import pandas as pd
4
 
5
  # Load the pre-trained model
6
- model = joblib.load("tuned_model.pkl")
7
 
8
  # Load the features used during training
9
  features = pd.read_csv("features_used_in_model.csv")["Feature"].tolist()
@@ -46,4 +46,6 @@ interface = gr.Interface(
46
  )
47
  )
48
 
49
- # Launch the interface for local testing or
 
 
 
3
  import pandas as pd
4
 
5
  # Load the pre-trained model
6
+ model = joblib.load("tuned_random_forest_model.pkl")
7
 
8
  # Load the features used during training
9
  features = pd.read_csv("features_used_in_model.csv")["Feature"].tolist()
 
46
  )
47
  )
48
 
49
+ # Launch the interface for local testing or Hugging Face Spaces deployment
50
+ if __name__ == "__main__":
51
+ interface.launch()