Update app.py

app.py

@@ -10,11 +10,26 @@ repo_id = "rasaenluis3/e3Modelo"
 learner = from_pretrained_fastai(repo_id)
 labels = learner.dls.vocab
 
+# Auxiliar
+def catToValue(cat):
+  if cat == '0':
+    return 'sadness'
+  elif cat == '1':
+    return 'joy' # wonderhoy :)
+  elif cat == '2':
+    return 'love'
+  elif cat == '3':
+    return 'anger'
+  elif cat == '4':
+    return 'fear'
+  elif cat == '5':
+    return 'surprise'
+
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(texto):
     #img = PILImage.create(img)
-    pred,pred_idx,probs = learner.predict(img)
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    pred,pred_idx,probs = learner.predict(texto)
+    return {catToValue(labels[i]): float(probs[i]) for i in range(len(labels))}
     
 # Creamos la interfaz y la lanzamos. 
 gr.Interface(fn=predict, inputs=gr.inputs.Textbox(lines=3,label="Escríbeme in english please"), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=False)