Update app.py

app.py

@@ -4,28 +4,13 @@ from huggingface_hub import from_pretrained_fastai
 
 # Cargar el modelo desde Hugging Face
 learn = from_pretrained_fastai("abelllanas/emotions_dl")
-
-# Crear un mapeo de etiquetas traducidas al inglés (solo una vez)
-label_map = {
-    "alegría cuadros pintura": "joy",
-    "ira cuadros pintura": "anger",
-    "miedo cuadros pintura": "fear",
-    "tristeza cuadros pintura": "sadness"
-}
+labels = learn.dls.vocab  # Obtener las etiquetas de emociones
 
 # Función para predecir emociones
 def predict(img):
     img = PILImage.create(img)  # Convertir la imagen al formato adecuado
     pred, pred_idx, probs = learn.predict(img)  # Realizar la predicción
-    
-    # Obtener las etiquetas originales
-    original_labels = learn.dls.vocab
-    
-    # Traducir las etiquetas utilizando el diccionario de mapeo
-    translated_labels = [label_map.get(label, label) for label in original_labels]
-    
-    # Devolver las probabilidades con las etiquetas traducidas
-    return {translated_labels[i]: float(probs[i]) for i in range(len(translated_labels))}
+    return {labels[i]: float(probs[i]) for i in range(len(labels))} 
 
 
 title = "Emotion predictor"