Update app.py

app.py

@@ -62,7 +62,21 @@ def main():
     # ... (rest of the code remains unchanged)
 
 
+    elif choice == "Upload Image":
+        st.subheader("Upload Your Own Images")
+        if "model" not in st.session_state:
+            st.warning("Please train the model first.")
+        else:
+            uploaded_files = st.file_uploader("Choose images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
+            if uploaded_files:
+                for img in uploaded_files:
+                    img = PILImage.create(img)
+                    label, _, probs = st.session_state.model.predict(img)
+
+                    st.image(img, caption=f"This is a {label}.")
+                    st.write(f"{label_a}: {probs[1].item():.6f}")
+                    st.write(f"{label_b}: {probs[0].item():.6f}")