diff --git a/app/api/api_outfit_matcher.py b/app/api/api_outfit_matcher.py index fb5bbe6..be507cb 100644 --- a/app/api/api_outfit_matcher.py +++ b/app/api/api_outfit_matcher.py @@ -32,12 +32,12 @@ def outfit_matcher(request_item: OutfitMatcher): best_outfits, best_scores = service.visualize(outfits, scores, request_item["topk"], best=True, # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_best_{param['topk']}.png") ) - result.append({"best_outfits": best_outfits, "best_scores": best_scores}) + result.append({"outfits": best_outfits, "scores": best_scores}) else: bad_outfits, bad_scores = service.visualize(outfits, scores, request_item["topk"], best=False, # output_path=os.path.join(r"E:\workspace\outfit_matcher\2024 SS Outfit", f"{item['item_name']}_worst_{param['topk']}.png") ) - result.append({"bad_outfits": bad_outfits, "bad_scores": bad_scores}) + result.append({"outfits": bad_outfits, "scores": bad_scores}) logger.info(f"run time is : {time.time() - start_time}") return {"message": "ok", "data": result} except Exception as e: