SonFox2920 commited on
Commit
0ef1031
·
verified ·
1 Parent(s): 139fda7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -13
app.py CHANGED
@@ -30,19 +30,6 @@ def preprocess_image(img_path):
30
  img = np.expand_dims(img, axis=0)
31
  return img
32
 
33
- # Function to get food information using Gemini
34
- def get_food_info(food_name, image_path):
35
- prompt = f"""Hình ảnh này chứa {food_name}, một món ăn Việt Nam. Dựa trên hình ảnh, hãy mô tả chi tiết món ăn này, bao gồm ngoại hình, nguyên liệu, và các đặc điểm đáng chú ý khác. Ngoài ra, liệt kê các đặc điểm chính của món ăn này. Trả về kết quả dưới dạng JSON: {{description: mô tả, features: [đặc điểm1, đặc điểm2, đặc điểm3, v.v.]}}"""
36
-
37
- img = Image.open(image_path)
38
- response = model.generate_content([prompt, img])
39
-
40
- try:
41
- food_info = json.loads(response.text)
42
- except json.JSONDecodeError:
43
- food_info = {"description": "Không thể tạo thông tin", "features": []}
44
-
45
- return food_info
46
 
47
 
48
  # Main function for the Streamlit app
 
30
  img = np.expand_dims(img, axis=0)
31
  return img
32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
 
34
 
35
  # Main function for the Streamlit app