jaifar530 commited on
Commit
c09fa65
·
unverified ·
1 Parent(s): c724997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -274,9 +274,9 @@ if press_me_button:
274
  file_prefix = 'not_trancated_full_paragraph.xlsx'
275
 
276
  # Load the models and vectorizer
277
- ridge_model = load_model(f"{file_prefix}_ridge_model.pkl")
278
- extra_trees_model = load_model(f"{file_prefix}_extra_trees_model.pkl")
279
- vectorizer = load_model(f"{file_prefix}_vectorizer.pkl")
280
  # Transform the input
281
  user_input_transformed = vectorizer.transform(new_text)
282
 
 
274
  file_prefix = 'not_trancated_full_paragraph.xlsx'
275
 
276
  # Load the models and vectorizer
277
+ ridge_model = pickle.load(f"{file_prefix}_ridge_model.pkl")
278
+ extra_trees_model = pickle.load(f"{file_prefix}_extra_trees_model.pkl")
279
+ vectorizer = pickle.load(f"{file_prefix}_vectorizer.pkl")
280
  # Transform the input
281
  user_input_transformed = vectorizer.transform(new_text)
282