Create app.py

app.py

@@ -0,0 +1,36 @@
+import streamlit as st
+import tensorflow as tf
+from tensorflow.keras.models import load_model
+from tensorflow.keras.preprocessing import image
+import numpy as np
+from PIL import Image
+
+# Load the model
+model = load_model('your_trained_model_resnet50.keras')
+
+# Streamlit title
+st.title("Tree Decoration Prediction Model")
+
+# Upload an image
+uploaded_file = st.file_uploader("Choose a tree image...", type="jpg")
+
+if uploaded_file is not None:
+    # Open the image file
+    img = Image.open(uploaded_file)
+    
+    # Preprocess the image for ResNet50 (resize, scale, etc.)
+    img = img.resize((224, 224))
+    img_array = np.array(img) / 255.0  # Normalize
+    img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
+    
+    # Make a prediction
+    prediction = model.predict(img_array)
+    
+    # Display prediction result
+    if prediction[0] > 0.5:
+        st.write("The tree is decorated!")
+    else:
+        st.write("The tree is undecorated.")
+    
+    # Display the image
+    st.image(img, caption="Uploaded Image.", use_column_width=True)