Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,8 +42,4 @@ async def predict(file: UploadFile = File(...)):
|
|
| 42 |
labels = model.config.id2label
|
| 43 |
result = {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 44 |
|
| 45 |
-
result = {
|
| 46 |
-
"female": float(probs[0]),
|
| 47 |
-
"male": float(probs[1])
|
| 48 |
-
}
|
| 49 |
return JSONResponse(content=result)
|
|
|
|
| 42 |
labels = model.config.id2label
|
| 43 |
result = {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
return JSONResponse(content=result)
|