Wills17 commited on
Commit
08225fa
·
verified ·
1 Parent(s): 7f54c7c

Update FastAPI_app.py

Browse files
Files changed (1) hide show
  1. FastAPI_app.py +4 -3
FastAPI_app.py CHANGED
@@ -227,9 +227,10 @@ async def detect_ingredients_hybrid(pil_image: Image.Image) -> List[Dict]:
227
  if name not in merged or detect["confidence"] > merged[name]["confidence"]:
228
  merged[name] = detect
229
 
230
- final = list(merged.values())
231
- return final[:] or [{"name": "No clear ingredients", "confidence": 0.0}]
232
-
 
233
 
234
  # Generate recipe with Qwen
235
  def generate_recipe_qwen(ingredient_names):
 
227
  if name not in merged or detect["confidence"] > merged[name]["confidence"]:
228
  merged[name] = detect
229
 
230
+ final_detections = list(merged.values())
231
+ # sort by confidence
232
+ final_detections.sort(key=lambda x: x["confidence"], reverse=True)
233
+ return final_detections or [{"name": "No clear ingredients", "confidence": 0.0}]
234
 
235
  # Generate recipe with Qwen
236
  def generate_recipe_qwen(ingredient_names):