Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,29 +3,31 @@ import gradio as gr
|
|
| 3 |
from transformers import pipeline
|
| 4 |
from huggingface_hub import login
|
| 5 |
|
| 6 |
-
# Autenticaci贸n
|
| 7 |
token = os.getenv("HF_TOKEN")
|
| 8 |
if token:
|
| 9 |
login(token=token)
|
| 10 |
|
| 11 |
-
#
|
| 12 |
-
|
|
|
|
|
|
|
| 13 |
try:
|
| 14 |
-
|
| 15 |
-
pipe = pipeline("text-classification", model=model_id, device="cpu")
|
| 16 |
except Exception as e:
|
|
|
|
| 17 |
print(f"Error cr铆tico al cargar el modelo: {e}")
|
| 18 |
|
| 19 |
def analizar_contenido(texto):
|
| 20 |
-
|
|
|
|
| 21 |
resultado = pipe(texto)
|
| 22 |
return resultado
|
| 23 |
|
| 24 |
-
# Interfaz
|
| 25 |
demo = gr.Interface(
|
| 26 |
fn=analizar_contenido,
|
| 27 |
-
inputs=gr.Textbox(label="Entrada de texto"
|
| 28 |
-
outputs=gr.JSON(label="Resultado
|
| 29 |
title="Analizador de Contenido"
|
| 30 |
)
|
| 31 |
|
|
|
|
| 3 |
from transformers import pipeline
|
| 4 |
from huggingface_hub import login
|
| 5 |
|
| 6 |
+
# Autenticaci贸n
|
| 7 |
token = os.getenv("HF_TOKEN")
|
| 8 |
if token:
|
| 9 |
login(token=token)
|
| 10 |
|
| 11 |
+
# CAMBIA ESTO por un modelo que s铆 tenga estructura de Transformers
|
| 12 |
+
# Ejemplo: Un clasificador de moderaci贸n de texto real
|
| 13 |
+
model_id = "s-nlp/roberta_toxicity_classifier"
|
| 14 |
+
|
| 15 |
try:
|
| 16 |
+
pipe = pipeline("text-classification", model=model_id)
|
|
|
|
| 17 |
except Exception as e:
|
| 18 |
+
pipe = None # Definimos pipe como None para evitar el NameError
|
| 19 |
print(f"Error cr铆tico al cargar el modelo: {e}")
|
| 20 |
|
| 21 |
def analizar_contenido(texto):
|
| 22 |
+
if pipe is None:
|
| 23 |
+
return "Error: El modelo no se carg贸 correctamente."
|
| 24 |
resultado = pipe(texto)
|
| 25 |
return resultado
|
| 26 |
|
|
|
|
| 27 |
demo = gr.Interface(
|
| 28 |
fn=analizar_contenido,
|
| 29 |
+
inputs=gr.Textbox(label="Entrada de texto"),
|
| 30 |
+
outputs=gr.JSON(label="Resultado"),
|
| 31 |
title="Analizador de Contenido"
|
| 32 |
)
|
| 33 |
|