Update food_identification.py

food_identification.py

@@ -1,16 +1,17 @@
-import json
+import tensorflow as tf  # Import TensorFlow
 import tensorflow_hub as hub
+import json
 import numpy as np
 
 # Load the Food101 model from TensorFlow Hub
 MODEL_URL = "https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/classification/5"
 model = hub.load(MODEL_URL)
 
-# Download and load Food-101 labels
+# Download and load ImageNet labels
 LABELS_URL = "https://storage.googleapis.com/download.tensorflow.org/data/imagenet_class_index.json"
 labels_path = tf.keras.utils.get_file("imagenet_labels.json", LABELS_URL)
 
-# Properly parse the JSON file
+# Properly parse the JSON file to load labels
 with open(labels_path, "r") as f:
     food_labels = json.load(f)
 
@@ -27,6 +28,6 @@ def identify_food(image):
     predictions = model(image_array)
     predicted_class = np.argmax(predictions[0])
 
-    # Get the top prediction
+    # Get the top prediction label
     food_item = food_labels.get(str(predicted_class), ["Unknown"])[1]
     return [food_item]