代码修改

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"]) outfits = fashion_dataset.generate_outfit(item, request_item["topk"], request_item["max_outfits"])
except ValueError as e: except ValueError as e:
logger.warning(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) 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}) result.append({"outfits": bad_outfits, "scores": bad_scores})
logger.info(f"run time is : {time.time() - start_time}") 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: # except Exception as e:
# logger.warning(e) # logger.warning(e)
# return {"message": f"{e}", "data": e} # return {"message": f"{e}", "data": e}