代码修改

This commit is contained in:
zhouchengrong
2024-04-03 15:27:15 +08:00
parent 0d598a4e11
commit 3b960afa83

View File

@@ -47,7 +47,7 @@ def outfit_matcher(request_item: OutfitMatcher):
outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])
except ValueError as e:
logger.warning(e)
return {"message": f"valueError : {e}", "data": e}
return {"code": 500, "message": f"valueError : {e}", "data": e}
scores = service.get_result(outfits, prepared_feature)
@@ -62,7 +62,7 @@ def outfit_matcher(request_item: OutfitMatcher):
)
result.append({"outfits": bad_outfits, "scores": bad_scores})
logger.info(f"run time is : {time.time() - start_time}")
return {"message": "ok", "data": result}
return {"code": 200, "message": "ok", "data": result}
# except Exception as e:
# logger.warning(e)
# return {"message": f"{e}", "data": e}