Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,8 +15,8 @@ label2id = {
|
|
15 |
}
|
16 |
# names of the files saved in step 2: Training
|
17 |
|
18 |
-
model_file_name = "
|
19 |
-
vectorizer_file_name = '
|
20 |
|
21 |
|
22 |
# load
|
|
|
15 |
}
|
16 |
# names of the files saved in step 2: Training
|
17 |
|
18 |
+
model_file_name = "xgb_reg_dg.pkl"
|
19 |
+
vectorizer_file_name = 'vectorizer_dg.pk'
|
20 |
|
21 |
|
22 |
# load
|