Midnightar commited on
Commit
fbd3022
·
verified ·
1 Parent(s): 8f136e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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)