Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,13 +10,12 @@ learner = from_pretrained_fastai(repo_id)
|
|
10 |
# Definimos una funci贸n que se encarga de llevar a cabo las predicciones
|
11 |
def predict(text):
|
12 |
pred,pred_idx,probs = learner.predict(text)
|
13 |
-
if (pred == "0")
|
14 |
return "Negative"
|
15 |
-
|
16 |
return "Positive"
|
17 |
-
|
18 |
return "Error"
|
19 |
-
}
|
20 |
|
21 |
# Creamos la interfaz y la lanzamos.
|
22 |
gr.Interface(fn=predict, inputs="text", outputs="label", examples=["I loved that movie", "I hated that movie"]).launch()
|
|
|
10 |
# Definimos una funci贸n que se encarga de llevar a cabo las predicciones
|
11 |
def predict(text):
|
12 |
pred,pred_idx,probs = learner.predict(text)
|
13 |
+
if (pred == "0"):
|
14 |
return "Negative"
|
15 |
+
elif (pred == "1"):
|
16 |
return "Positive"
|
17 |
+
else:
|
18 |
return "Error"
|
|
|
19 |
|
20 |
# Creamos la interfaz y la lanzamos.
|
21 |
gr.Interface(fn=predict, inputs="text", outputs="label", examples=["I loved that movie", "I hated that movie"]).launch()
|