Update app.py

app.py

@@ -3,6 +3,8 @@
 import streamlit as st
 import base64
 from main import generate_and_upscale_image
+st.set_option('server.maxMessageSize', 400)  # Increase to 300MB
+
 
 def get_image_download_link(img_bytes, filename):
     b64 = base64.b64encode(img_bytes).decode()