Dannel commited on
Commit
364c747
verified
1 Parent(s): 9918935

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -27
app.py CHANGED
@@ -1,32 +1,14 @@
1
- import gradio as gr
2
  from transformers import pipeline
 
3
 
4
- # Cargar el modelo pre-entrenado
5
- gender_classifier = pipeline('text-classification', model='Dannel/Gender_Classifier')
6
 
7
- def infer_gender(name):
8
- """
9
- Infiere el g茅nero de una persona a partir de su nombre.
10
-
11
- Args:
12
- name (str): El nombre de la persona.
13
-
14
- Returns:
15
- str: El g茅nero predicho ('Male' o 'Female').
16
- """
17
- # Hacer la predicci贸n utilizando el modelo cargado
18
- prediction = gender_classifier([name])[0]
19
-
20
- return prediction['label']
21
 
22
- # Crear la interfaz de Gradio
23
- demo = gr.Interface(
24
- fn=infer_gender,
25
- inputs=gr.Textbox(label="Nombre"),
26
- outputs=gr.Label(label="G茅nero predicho"),
27
- title="Clasificador de G茅nero",
28
- description="Ingresa un nombre para predecir su g茅nero."
29
- )
30
 
31
- # Ejecutar la aplicaci贸n
32
- demo.launch(share=True)
 
1
+ import os
2
  from transformers import pipeline
3
+ from huggingface_hub import HfApi
4
 
5
+ # Obt茅n la clave de acceso de la variable de entorno
6
+ access_token = os.environ.get("HUGGING_FACE_ACCESS_TOKEN")
7
 
8
+ # Crear una instancia de HfApi
9
+ api = HfApi(endpoint="https://huggingface.co")
 
 
 
 
 
 
 
 
 
 
 
 
10
 
11
+ # Cargar el modelo pre-entrenado con autenticaci贸n
12
+ gender_classifier = pipeline('text-classification', model='Dannel/Gender_Classifier', use_auth_token=access_token)
 
 
 
 
 
 
13
 
14
+ # Resto del c贸digo...