Update app.py

app.py

@@ -104,7 +104,7 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
             return json.loads(response.decode())[0]["generated_text"]
         
         # Use the prediction to generate nutrition information
-        prompt = f"Nutrition information for {prediction} in formatted list"
+        prompt = f"Nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} in formatted list"
         response = call_llm(llm_client, prompt)
 
         return {"predicted_label": prediction, "nutrition_info": response}