Update app.py

app.py

@@ -61,3 +61,8 @@ import os
 if st.button("Train Model"):
     os.system("python train_model.py")
     st.success("Model training complete. Saved as deep_learning_model.h5")
+
+if not os.path.exists("deep_learning_model.h5"):
+    st.info("Training the model for the first time...")
+    os.system("python train_model.py")
+    st.success("Model trained successfully and saved as deep_learning_model.h5")