allopeap commited on
Commit
7b62b61
verified
1 Parent(s): 88e6f6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
- from fastai.text.all import *
4
 
5
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
6
  repo_id = "allopeap/emociones"
@@ -9,8 +9,8 @@ learner = from_pretrained_fastai(repo_id)
9
  labels = learner.dls.vocab
10
 
11
  # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
12
- def predict(txt):
13
- pred,_, probs = learner.predict(txt)
14
  if pred == '0':
15
  return "Positive"
16
  elif pred == 1:
 
1
  from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
+ from fastai.vision.all import *
4
 
5
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
6
  repo_id = "allopeap/emociones"
 
9
  labels = learner.dls.vocab
10
 
11
  # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
12
+ def predict(text):
13
+ pred,pred_idx, probs = learner.predict(text)
14
  if pred == '0':
15
  return "Positive"
16
  elif pred == 1: