juancamval commited on
Commit
a1ff76a
verified
1 Parent(s): 90fb0c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -4,10 +4,10 @@ from transformers import pipeline
4
  import os
5
 
6
  # Define las rutas absolutas a tus modelos
7
- MODEL_TRANSFORMER_ENCODER_PATH = '/stacked_transformer_encoder.keras'
8
- MODEL_TRANSFORMER_POS_PATH = '/transformer_encoder_pos.keras'
9
- MODEL_SIMPLE_RNN_PATH = '/model_simple_rnn.keras'
10
- MODEL_LSTM_PATH = '/model_lstm.keras'
11
 
12
  # Cargar tus modelos de clasificaci贸n forzando la ruta absoluta
13
  try:
@@ -35,7 +35,7 @@ except Exception as e:
35
  # Cargar el pipeline de traducci贸n de Hugging Face (ingl茅s a espa帽ol)
36
  try:
37
  hf_token = os.environ.get("HF_TOKEN")
38
- translator_en_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es", token=hf_token)
39
  st.info("Modelo de traducci贸n (ingl茅s a espa帽ol) cargado correctamente con token.")
40
  except Exception as e:
41
  st.error(f"Error al cargar el modelo de traducci贸n (ingl茅s a espa帽ol): {e}")
 
4
  import os
5
 
6
  # Define las rutas absolutas a tus modelos
7
+ MODEL_TRANSFORMER_ENCODER_PATH = 'stacked_transformer_encoder.keras'
8
+ MODEL_TRANSFORMER_POS_PATH = 'transformer_encoder_pos.keras'
9
+ MODEL_SIMPLE_RNN_PATH = 'model_simple_rnn.keras'
10
+ MODEL_LSTM_PATH = 'model_lstm.keras'
11
 
12
  # Cargar tus modelos de clasificaci贸n forzando la ruta absoluta
13
  try:
 
35
  # Cargar el pipeline de traducci贸n de Hugging Face (ingl茅s a espa帽ol)
36
  try:
37
  hf_token = os.environ.get("HF_TOKEN")
38
+ modelo_traduccion_en_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es",token = hf_token)
39
  st.info("Modelo de traducci贸n (ingl茅s a espa帽ol) cargado correctamente con token.")
40
  except Exception as e:
41
  st.error(f"Error al cargar el modelo de traducci贸n (ingl茅s a espa帽ol): {e}")