Spaces:
Sleeping
Sleeping
jaifar530
commited on
Update app.py
Browse files
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
|
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 |
|