jaifar530 commited on
Commit
e378435
·
unverified ·
1 Parent(s): 454e5fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -4
app.py CHANGED
@@ -10,7 +10,7 @@ from sklearn.preprocessing import LabelEncoder
10
  import pickle
11
  import numpy as np
12
  from PIL import Image
13
- from file_checker import check_and_download_files
14
 
15
  st.write(f"Pickle version: {pickle.format_version}")
16
 
@@ -275,10 +275,30 @@ if press_me_button:
275
  file_prefix = 'not_trancated_full_paragraph.xlsx'
276
 
277
  # Load the models and vectorizer
278
- ridge_model = load_model(f"{file_prefix}_ridge_model.pkl")
279
- extra_trees_model = load_model(f"{file_prefix}_extra_trees_model.pkl")
280
- vectorizer = load_model(f"{file_prefix}_vectorizer.pkl")
281
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
282
  # Transform the input
283
  user_input_transformed = vectorizer.transform([new_text])
284
 
 
10
  import pickle
11
  import numpy as np
12
  from PIL import Image
13
+ from joblib import load
14
 
15
  st.write(f"Pickle version: {pickle.format_version}")
16
 
 
275
  file_prefix = 'not_trancated_full_paragraph.xlsx'
276
 
277
  # Load the models and vectorizer
278
+ #ridge_model = load_model(f"{file_prefix}_ridge_model.pkl")
 
 
279
 
280
+ try:
281
+ ridge_model = load(f"{file_prefix}_ridge_model.pkl")
282
+ except Exception as e:
283
+ st.write(f"Error loading the ridge model: {e}")
284
+ exit(1)
285
+
286
+ # extra_trees_model = load_model(f"{file_prefix}_extra_trees_model.pkl")
287
+ try:
288
+ extra_trees_model = load(f"{file_prefix}_extra_trees.pkl")
289
+ except Exception as e:
290
+ st.write(f"Error loading the extra_trees: {e}")
291
+ exit(1)
292
+
293
+ # vectorizer = load_model(f"{file_prefix}_vectorizer.pkl")
294
+
295
+ try:
296
+ with open(f"{file_prefix}_vectorizer.pkl", 'rb') as f:
297
+ vectorizer = pickle.load(f)
298
+ except Exception as e:
299
+ st.write(f"Error loading the vectorizer: {e}")
300
+ exit(1)
301
+
302
  # Transform the input
303
  user_input_transformed = vectorizer.transform([new_text])
304