mapecim commited on
Commit
c41508a
1 Parent(s): 2b9e59d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
- } elif (pred == "1") {
16
  return "Positive"
17
- } else {
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()