thak123 commited on
Commit
50ca09f
1 Parent(s): dbc06ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ label2id = {
15
  }
16
  # names of the files saved in step 2: Training
17
 
18
- model_file_name = "xgb_reg.pkl"
19
- vectorizer_file_name = 'vectorizer.pk'
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