allopeap commited on
Commit
1cbdfbe
verified
1 Parent(s): d817748

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from huggingface_hub import from_pretrained_fastai
2
+ import gradio as gr
3
+ from fastai.text.all import *
4
+
5
+
6
+ # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
7
+ repo_id = "allopeap/emociones"
8
+
9
+ learner = from_pretrained_fastai(repo_id)
10
+ labels = learner.dls.vocab
11
+
12
+ # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
13
+ def predict(txt):
14
+ #img = PILImage.create(img)
15
+ pred,pred_idx,probs = learner.predict(txt)
16
+ return {labels[i]: float(probs[i]) for i in range(len(labels))}
17
+
18
+ # Creamos la interfaz y la lanzamos.
19
+ gr.Interface(
20
+ fn=predict,
21
+ inputs=gr.inputs.Textbox(lines=2, placeholder="Ingresa el texto aqu铆..."),
22
+ outputs=gr.outputs.Label(num_top_classes=3),
23
+ examples=[['Necesito seguir a mas cuentas camren shippers y fans de las armon铆as. Me recomendais alguna?'], ['La felicidad tiene un nombr #Tailandia Si es tu PRIMERA VEZ, apunta estos #consejos! '], ['Todo el mundo con la segunda temporada de Narcos y yo no, PORQUE NO TENGO NETFLIX @user pasaros una suscripcion de 5 dias']]
24
+ ).launch(share=False)