Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|