Fix Gradio 6 css/js injection

app.py

@@ -7,6 +7,7 @@ import time
 import gradio as gr
 
 MAX_DEMO_LEN = 20
+GRADIO_MAJOR = int((gr.__version__ or "0").split(".", maxsplit=1)[0])
 
 ROSA_CODE = """
 def rosa_qkv(qqq, kkk, vvv):
@@ -1450,7 +1451,12 @@ JS = """
 """
 
 
-with gr.Blocks(css=CSS, js=JS) as demo:
+if GRADIO_MAJOR >= 6:
+    demo_context = gr.Blocks()
+else:
+    demo_context = gr.Blocks(css=CSS, js=JS)
+
+with demo_context as demo:
     gr.HTML(
         '<div class="page-header">'
         '<div class="page-title">ROSA QKV Demo</div>'
@@ -1493,4 +1499,7 @@ with gr.Blocks(css=CSS, js=JS) as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    launch_kwargs = {}
+    if GRADIO_MAJOR >= 6:
+        launch_kwargs = {"css": CSS, "js": JS, "ssr_mode": False}
+    demo.launch(**launch_kwargs)