Update app.py

app.py

@@ -3,6 +3,9 @@ from huggingface_hub import InferenceClient
 from gradio_client import Client
 import re
 
+# Set page configuration
+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)
@@ -87,7 +90,6 @@ def generate_image(prompt):
         return None
 
 def main():
-    st.set_page_config(layout="wide")
     st.title("Free Waifu Character Generator")
     st.write("""
     Welcome to the Free Waifu Character Generator! This app allows you to create a unique waifu with just one click.