Update app.py

app.py

@@ -5,17 +5,16 @@ import numpy as np
 import zipfile
 import os
 
-
 # Function to load the model from the zip file
 def load_model_from_zip(zip_file_path):
     with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
         zip_ref.extractall('.')  # Extract to the current directory
-    # Load the SavedModel (no need for model_path anymore)
-    model = tf.keras.models.load_model('your_trained_model')  # Replace 'your_trained_model' with your model folder name, if any
+    # Load the SavedModel directly from the current directory (root)
+    model = tf.keras.models.load_model('.')  
     return model
 
-# Load the model
-model = load_model_from_zip('my_christmas_tree_model.zip')  # Update with your actual zip file name
+# Load the model 
+model = load_model_from_zip('my_christmas_tree_model.zip')  
 
 # Streamlit UI
 st.title("Christmas Tree Classifier")