J007acky commited on
Commit
0676242
·
verified ·
1 Parent(s): 13f784a

Update FastAPI.py

Browse files
Files changed (1) hide show
  1. FastAPI.py +2 -2
FastAPI.py CHANGED
@@ -82,10 +82,10 @@ def recognize_face(image_url: HttpUrl) -> ImgOutput:
82
  os.remove("examp.jpg")
83
 
84
  if output_labels:
85
- return ImgOutput(label=output_labels)
86
  else:
87
  out = ["unable to detect"]
88
- return ImgOutput(label=out)
89
 
90
 
91
 
 
82
  os.remove("examp.jpg")
83
 
84
  if output_labels:
85
+ return ImgOutput(user_id=output_labels)
86
  else:
87
  out = ["unable to detect"]
88
+ return ImgOutput(user_id=out)
89
 
90
 
91