Update app.py

app.py

@@ -17,9 +17,13 @@ from tensorflow.keras.datasets import mnist,cifar10
 
 option = st.selectbox(
     "Which model would you like to get prediction with?",
-    ("Auto-Regressor", "Auto-Encoder", "Diffusion-Model","Other"))
+    ("None","Auto-Regressor", "Auto-Encoder", "Diffusion-Model","Other"))
 
 st.write("You selected:", option)
+
+if option == "None":
+    st.write("Please Select the model to get the fun prediction.... :)")
+
 if option == "Auto-Encoder":
     st.write("It is under development")
     st.write("Stay tune... Comming soon... :)")
@@ -34,6 +38,9 @@ if option == "Diffusion-Model":
 if option == "Auto-Regressor":
     if st.button("Run"):
         st.write("Running Auto-Regressor")
+
+        st.write("trained on --> cifar-10 dataset, RTX-GPU's, 50-epochs")
+        st.write("This is trail model, updated version will be updated consicutively.")
         
         (trainX, trainy), (testX, testy) = cifar10.load_data()