Spaces:
Running
Running
Update FastAPI_app.py
Browse files- 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 |
-
|
| 231 |
-
|
| 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):
|