appinitdev commited on
Commit
3009ecf
verified
1 Parent(s): b7f9a59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -9
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 con el token secreto configurado en el Space
7
  token = os.getenv("HF_TOKEN")
8
  if token:
9
  login(token=token)
10
 
11
- # Carga del modelo
12
- model_id = "Bigfawg6669/create-nsfw"
 
 
13
  try:
14
- # Usamos pipeline para simplificar la clasificaci贸n
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
- # La 'entrada' es el texto que el usuario escribe en la caja de Gradio
 
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", placeholder="Escribe aqu铆 para clasificar..."),
28
- outputs=gr.JSON(label="Resultado del modelo"),
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