Spaces:
Sleeping
Sleeping
Commit ·
fd07503
1
Parent(s): 8d4faa9
update label
Browse files
app.py
CHANGED
|
@@ -83,9 +83,7 @@ def predict(
|
|
| 83 |
]
|
| 84 |
)
|
| 85 |
pred = rf.predict(data)[0]
|
| 86 |
-
|
| 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 = [
|