Update app.py

app.py

@@ -119,12 +119,21 @@ CSS="""
 .output { width=112px; height=112px; max_width=112px; max_height=112px; !important; }
 .gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
 .guide { text-align: center; !important; }
-/* TESTING */
 """
 
+js_func = """
+function refresh() {
+    const url = new URL(window.location);
 
-with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
-# with gr.Blocks(theme='JohnSmith9982/small_and_pretty', fill_width=True, css=CSS) as demo:
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
+
+# with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS, js=js_func) as demo:
+with gr.Blocks(theme='JohnSmith9982/small_and_pretty', fill_width=True, css=CSS) as demo:
     gr.HTML("")  
     with gr.Tab('6 Models'):
         with gr.Column(scale=2):