bego commited on
Commit
acc1f6f
1 Parent(s): d434e51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import pickle
3
  import ftfy
4
  import re
5
  import torch
@@ -22,12 +22,9 @@ class ClasificadorOpiniones:
22
  def __init__(self):
23
  try:
24
  model_path = hf_hub_download(repo_id="begoach1/opinion_classifier", filename="modelo_clasificador_reentrenado_lp_ros.pkl")
25
- with open(model_path, 'rb') as f:
26
- self.clf_combined = pickle.load(f)
27
  except FileNotFoundError:
28
  raise RuntimeError("El archivo del modelo no se encuentra.")
29
- except pickle.UnpicklingError as e:
30
- raise RuntimeError(f"Error al deserializar el modelo. Asegúrate de que el archivo no esté corrupto y que la versión de scikit-learn sea compatible. Detalles: {e}")
31
  except Exception as e:
32
  raise RuntimeError(f"Error al descargar o cargar el modelo: {e}")
33
 
 
1
  import gradio as gr
2
+ import joblib
3
  import ftfy
4
  import re
5
  import torch
 
22
  def __init__(self):
23
  try:
24
  model_path = hf_hub_download(repo_id="begoach1/opinion_classifier", filename="modelo_clasificador_reentrenado_lp_ros.pkl")
25
+ self.clf_combined = joblib.load(model_path)
 
26
  except FileNotFoundError:
27
  raise RuntimeError("El archivo del modelo no se encuentra.")
 
 
28
  except Exception as e:
29
  raise RuntimeError(f"Error al descargar o cargar el modelo: {e}")
30