Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,10 +2,10 @@ import os
 os.environ["TF_USE_LEGACY_KERAS"] = "1"
 
 import streamlit as st
-import tensorflow as tf
-from tensorflow.keras.applications.efficientnet import preprocess_input
-from PIL import Image
 import numpy as np
+from tensorflow import keras
+from keras.applications.efficientnet import preprocess_input
+from PIL import Image
 
 # -----------------------------------
 # CONFIG
@@ -18,10 +18,7 @@ CLASS_NAMES = ['cup', 'fork', 'glass', 'knife', 'plate', 'spoon']
 # -----------------------------------
 @st.cache_resource
 def load_model_file():
-    return tf.keras.models.load_model(
-        "src/kitchen1.keras",
-        compile=False
-    )
+    return keras.models.load_model("src/kitchen1.keras", compile=False)
 
 model = load_model_file()