Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from huggingface_hub import from_pretrained_fastai
|
2 |
import gradio as gr
|
3 |
-
from fastai.
|
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(
|
13 |
-
pred,
|
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:
|