juancamval commited on
Commit
f850537
verified
1 Parent(s): 77fc659

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -3,7 +3,7 @@ import tensorflow as tf
3
  from transformers import pipeline
4
  import os
5
 
6
- # Cargar tus modelos de clasificaci贸n (asumiendo que est谩n en la ra铆z del proyecto)
7
  try:
8
  model_transformer_encoder = tf.keras.models.load_model('stacked_transformer_encoder.keras')
9
  model_transformer_positional_encoding = tf.keras.models.load_model('transformer_encoder_pos.keras')
@@ -17,7 +17,7 @@ except Exception as e:
17
  model_simple_rnn = None
18
  model_lstm = None
19
 
20
- # Cargar el pipeline de traducci贸n de Hugging Face (ingl茅s a espa帽ol)
21
  try:
22
  translator_en_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
23
  except Exception as e:
@@ -37,12 +37,8 @@ def clasificar_noticia(texto, modelo_seleccionado):
37
  modelo_a_usar = model_lstm
38
 
39
  if modelo_a_usar:
40
- # *** IMPORTANTE: Implementa aqu铆 la l贸gica de preprocesamiento del texto ***
41
- # para que coincida con la entrada esperada por el modelo.
42
- # Esto incluye tokenizaci贸n, padding, etc.
43
  prediction = modelo_a_usar.predict([texto])
44
- # *** IMPORTANTE: Interpreta la salida del modelo para obtener la clase predicha. ***
45
- # Esto depender谩 de c贸mo entrenaste tu modelo.
46
  return f"Clase predicha ({modelo_seleccionado}): {prediction}"
47
  else:
48
  return f"El modelo '{modelo_seleccionado}' no est谩 disponible."
 
3
  from transformers import pipeline
4
  import os
5
 
6
+
7
  try:
8
  model_transformer_encoder = tf.keras.models.load_model('stacked_transformer_encoder.keras')
9
  model_transformer_positional_encoding = tf.keras.models.load_model('transformer_encoder_pos.keras')
 
17
  model_simple_rnn = None
18
  model_lstm = None
19
 
20
+ # Cargar el pipeline de traducci贸n de Hugging Face
21
  try:
22
  translator_en_es = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
23
  except Exception as e:
 
37
  modelo_a_usar = model_lstm
38
 
39
  if modelo_a_usar:
40
+
 
 
41
  prediction = modelo_a_usar.predict([texto])
 
 
42
  return f"Clase predicha ({modelo_seleccionado}): {prediction}"
43
  else:
44
  return f"El modelo '{modelo_seleccionado}' no est谩 disponible."