Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import tensorflow as tf
|
|
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,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
|
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 |
-
|
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."
|