Update app.py

app.py

@@ -11,7 +11,7 @@ def predict(image):
     return {p["label"]: p["score"] for p in predictions}
 
 def main():
-    st.title("Image Classification")
+    st.title("Age Classification From Image")
     
     with st.form("my_form"):
         uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
@@ -19,8 +19,8 @@ def main():
         if uploaded_file is not None:
         # Display the uploaded image
             image = Image.open(uploaded_file)
-            st.image(image, caption="Uploaded Image", use_column_width=True)
-        clicked = st.form_submit_button("Predict")
+            st.image(image, caption="Your uploaded Image", use_column_width=True)
+        clicked = st.form_submit_button("Press to predict")
         if clicked:
             results = predict(image)
             k = [] 
@@ -31,7 +31,7 @@ def main():
                 k.append(key)
             vp = [str(item) + '%' for item in v]
             result = k[0]
-            st.success('The predicted image is {}'.format(result))
+            st.success('The predicted age is {}'.format(result))
             df = pd.DataFrame({'Prediction': k,'Accuracy':vp})
             st.dataframe(df,hide_index=True)