Update app.py

app.py

@@ -3,6 +3,9 @@ from huggingface_hub import InferenceClient
 from gradio_client import Client
 import re
 
+# Set the page config
+st.set_page_config(layout="wide")
+
 # Load custom CSS
 with open('style.css') as f:
     st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
@@ -83,7 +86,6 @@ def generate_image(prompt):
         return None
 
 def main():
-    st.set_page_config(layout="wide")
     st.title("Enhanced Waifu Character Generator")
 
     # User inputs