Update app.py

app.py

@@ -5,5 +5,12 @@ import keras
 
 model = keras.models.load_model("model.keras")
 
+old_stdout = sys.stdout  # Save the current stdout so we can restore it later
+sys.stdout = StringIO()  # Redirect stdout to StringIO object
+model.summary()  # This will now be written to the StringIO object
+model_summary = sys.stdout.getvalue()  # Get the value from StringIO
+sys.stdout = old_stdout  # Restore original stdout
+
+# Display model summary using Streamlit
 st.write("### Model Summary:")
-st.text(str(model.summary()))
+st.text(model_summary)