ChatbotML commited on
Commit
3439baf
verified
1 Parent(s): f8f9017

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -0
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from transformers import pipeline
4
+
5
+ pipe = pipeline("image-classification",
6
+ model="dima806/facial_emotions_image_detection")
7
+
8
+ def launch(input):
9
+ etiquetas_espanol = {'happy': 'felicidad', 'neutral': 'neutral',
10
+ 'surprise': 'sorpresa', 'sad': 'tristeza',
11
+ 'angry': 'ira', 'fear': 'miedo',
12
+ 'disgust': 'aversi贸n'}
13
+
14
+ emocion_puntajes = pipe(input)
15
+
16
+ cadenas_formateadas = []
17
+
18
+ for emocion in emocion_puntajes:
19
+ label = emocion['label']
20
+ score = emocion['score']
21
+ palabra_espanol = etiquetas_espanol[label]
22
+ porcentaje = round(score * 100, 1)
23
+ cadena_formateada = f"{palabra_espanol}: {porcentaje}%"
24
+ cadenas_formateadas.append(cadena_formateada)
25
+
26
+ # Unir las cadenas formateadas con comas y un espacio
27
+ resultado = ", ".join(cadenas_formateadas)
28
+ return resultado
29
+
30
+ iface = gr.Interface(launch,
31
+ inputs=gr.Image(type='pil'),
32
+ outputs="text")
33
+
34
+ iface.launch()