Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import lightgbm as lgb
|
|
4 |
import pickle
|
5 |
|
6 |
# Load the trained LightGBM model
|
7 |
-
with open('
|
8 |
model = pickle.load(model_file)
|
9 |
|
10 |
# Define mappings
|
|
|
4 |
import pickle
|
5 |
|
6 |
# Load the trained LightGBM model
|
7 |
+
with open('lightgbm_model.pkl', 'rb') as model_file:
|
8 |
model = pickle.load(model_file)
|
9 |
|
10 |
# Define mappings
|