Update app.py

app.py

@@ -55,3 +55,9 @@ if st.button("Detect"):
         st.write(f"**Confidence Score:** {confidence:.2f}%")
     else:
         st.warning("Please enter a prompt to test.")
+        
+import os
+
+if st.button("Train Model"):
+    os.system("python train_model.py")
+    st.success("Model training complete. Saved as deep_learning_model.h5")