Lord-Raven commited on
Commit ·
260fdda
1
Parent(s): 1b99eaa
Trying to fix parameter issue.
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ classifier = pipeline(
|
|
| 21 |
def predict(param_0):
|
| 22 |
results = classifier(param_0)
|
| 23 |
response = {
|
| 24 |
-
"label": results[0][0]
|
| 25 |
"confidences": results[0]
|
| 26 |
}
|
| 27 |
return response
|
|
|
|
| 21 |
def predict(param_0):
|
| 22 |
results = classifier(param_0)
|
| 23 |
response = {
|
| 24 |
+
"label": results[0][0]["label"],
|
| 25 |
"confidences": results[0]
|
| 26 |
}
|
| 27 |
return response
|