Update app.py

app.py

@@ -1,22 +1,32 @@
 import gradio as gr
 import os
-
-# Force Keras 2 behavior before importing tensorflow
+# Force legacy Keras
 os.environ["TF_USE_LEGACY_KERAS"] = "1"
 
+import gradio as gr
 import tensorflow as tf
+import tf_keras as keras  # Use this instead of tf.keras
 import numpy as np
 from PIL import Image
 
-# Use the legacy loader if you are on TF 2.16+
-try:
-    import tf_keras as keras
-except ImportError:
-    from tensorflow import keras
-
+# 1. Load your model using the tf_keras loader
 MODEL_PATH = "model/FOOD CAL MODEL.h5"
 model = keras.models.load_model(MODEL_PATH)
 
+# ... (rest of your labels and NUTRITION_DB here) ...
+
+def predict_nutrition(img):
+    if img is None:
+        return None, "Please upload an image."
+    
+    # Preprocessing
+    img = Image.fromarray(img).resize((224, 224))
+    img_array = keras.preprocessing.image.img_to_array(img)
+    img_array = np.expand_dims(img_array, axis=0) / 255.0
+
+    # Prediction
+    prediction = model.predict(img_array)[0]
+
 # 2. Labels & Database (Make sure to include the full dictionary from previous turns)
 LABELS = [
     'apple_pie', 'baby_back_ribs', 'baklava', 'beef_carpaccio', 'beef_tartare', 'beet_salad', 'beignets',