Spaces:
Running
Running
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)
|