Update app.py

app.py

@@ -19,6 +19,16 @@ from transformers import (
     TextIteratorStreamer,
 )
 
+js_func = """
+function refresh() {
+    const url = new URL(window.location);
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
+
 # --- Constants and Model Setup ---
 MAX_INPUT_TOKEN_LENGTH = 4096
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -166,7 +176,7 @@ def create_gradio_interface():
     .process-button { border: none !important; color: white !important; font-weight: bold !important; background-color: blue !important;} 
     .process-button:hover { background-color: darkblue !important; transform: translateY(-2px) !important; box-shadow: 0 4px 8px rgba(0,0,0,0.2) !important; }
     """
-    with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
+    with gr.Blocks(theme="bethecloud/storj_theme", css=css, js=js_func) as demo:
         gr.HTML("""
         <div class="title" style="text-align: center">
             <h1>Dot<span style="color: red;">●</span><strong></strong>OCR Comparator</h1>