1MR commited on
Commit
8e98de4
·
verified ·
1 Parent(s): fbbc76e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -114,8 +114,8 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
114
  # return {"predicted_label": prediction, "nutrition_info": response}
115
 
116
  # nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} per 100 grams in a formatted list only."
117
- nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6) for {prediction} per 100 grams, Output the information as a concise, formatted list without repetition."
118
- nutrition_info = call_llm(llm_client, nutrition_prompt)
119
 
120
  # # Second prompt: Health benefits and tips
121
  health_benefits_prompt = f"Provide the health benefits and considerations for {prediction}. Additionally, include practical tips for making {prediction} healthier. Keep the response focused on these two aspects only."
@@ -124,7 +124,7 @@ async def predict_image_and_nutrition(file: UploadFile = File(...)):
124
  return {
125
  "predicted_label": prediction,
126
  "health_benefits_and_tips": health_benefits_and_tips,
127
- "nutrition_info": nutrition_info
128
  }
129
  except Exception as e:
130
  return JSONResponse(
 
114
  # return {"predicted_label": prediction, "nutrition_info": response}
115
 
116
  # nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6, Folate, Niacin, Pantothenic acid) for {prediction} per 100 grams in a formatted list only."
117
+ # nutrition_prompt = f"Provide the nutrition information (Calories, Protein, Carbohydrates, Dietary Fiber, Sugars, Fat, Sodium, Potassium, Vitamin C, Vitamin B6) for {prediction} per 100 grams, Output the information as a concise, formatted list without repetition."
118
+ # nutrition_info = call_llm(llm_client, nutrition_prompt)
119
 
120
  # # Second prompt: Health benefits and tips
121
  health_benefits_prompt = f"Provide the health benefits and considerations for {prediction}. Additionally, include practical tips for making {prediction} healthier. Keep the response focused on these two aspects only."
 
124
  return {
125
  "predicted_label": prediction,
126
  "health_benefits_and_tips": health_benefits_and_tips,
127
+ # "nutrition_info": nutrition_info
128
  }
129
  except Exception as e:
130
  return JSONResponse(