bego commited on
Commit
dbe5dae
verified
1 Parent(s): f7ac5c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -30,8 +30,17 @@ def preprocesar_texto(texto):
30
 
31
  class ClasificadorOpiniones:
32
  def __init__(self):
33
- model_path = hf_hub_download(repo_id="begoach1/opinion_classifier", filename="modelo_clasificador_reentrenado_lp_ros.pkl", use_auth_token=hf_token)
34
- self.clf_combined = pickle.load(open(model_path, 'rb'))
 
 
 
 
 
 
 
 
 
35
  self.tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-multilingual-cased')
36
  self.model = DistilBertModel.from_pretrained('distilbert-base-multilingual-cased')
37
 
@@ -60,4 +69,4 @@ iface = gr.Interface(
60
  description="Ingresa un texto de opini贸n para obtener las etiquetas correspondientes."
61
  )
62
 
63
- iface.launch(share=True)
 
30
 
31
  class ClasificadorOpiniones:
32
  def __init__(self):
33
+ try:
34
+ model_path = hf_hub_download(repo_id="begoach1/opinion_classifier", filename="modelo_clasificador_reentrenado_lp_ros.pkl", use_auth_token=hf_token)
35
+ with open(model_path, 'rb') as f:
36
+ self.clf_combined = pickle.load(f)
37
+ except FileNotFoundError:
38
+ raise RuntimeError("El archivo del modelo no se encuentra.")
39
+ except pickle.UnpicklingError:
40
+ raise RuntimeError("Error al deserializar el modelo. Aseg煤rate de que el archivo no est茅 corrupto y que la versi贸n de scikit-learn sea compatible.")
41
+ except Exception as e:
42
+ raise RuntimeError(f"Error al descargar o cargar el modelo: {e}")
43
+
44
  self.tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-multilingual-cased')
45
  self.model = DistilBertModel.from_pretrained('distilbert-base-multilingual-cased')
46
 
 
69
  description="Ingresa un texto de opini贸n para obtener las etiquetas correspondientes."
70
  )
71
 
72
+ iface.launch()