revert

app.py

@@ -341,8 +341,8 @@ example_prompts = [
 
 # Use a state variable to store session ID
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    # session_id = gr.State(None)  # Add this to store session ID
-    session_id = gr.BrowserState()(None)  # Add this to store session ID
+    session_id = gr.State(None)  # Add this to store session ID
+    # session_id = gr.BrowserState()(None)  # Add this to store session ID
 
     gr.Markdown("# 🌊 WaveSpeedAI HiDream Arena")
 

requirements.txt

@@ -1,4 +1,4 @@
-gradio==5.25.0
+gradio
 aiohttp
 plotly
 python-dotenv