luismarques commited on
Commit
fd07503
·
1 Parent(s): 8d4faa9

update label

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -83,9 +83,7 @@ def predict(
83
  ]
84
  )
85
  pred = rf.predict(data)[0]
86
- print(pred)
87
- pred_proba = rf.predict_proba(data)[0][0]
88
- return {"label": pred, "confidence": pred_proba}
89
 
90
 
91
  inputs = [
 
83
  ]
84
  )
85
  pred = rf.predict(data)[0]
86
+ return {"fumante": pred}
 
 
87
 
88
 
89
  inputs = [