Update app.py

app.py

@@ -65,21 +65,32 @@ def predict(img):
     probs, indices = probs.tolist(), indices.tolist()  # transform to list
     return {LABELS[i]: float(v) for i, v in zip(indices, probs)}
 
+js_func = """
+function refresh() {
+    const url = new URL(window.location);
 
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Sketchpad(
-        label="Draw Here",
-        brush=gr.Brush(default_size=20, default_color="#FFFFFF", colors=["#FFFFFF"]),
-        image_mode="L",
-        layers=False,
-        eraser=None,
-    ),
-    outputs=gr.Label(label="Guess"),
-    title="Thai Digit Handwritten Classification",
-    description="ทดลองวาดภาพตัวอักษรเลขไทยลงใน Sketchpad ด้านล่างเพื่อทำนายผลตัวเลข ตั้งแต่ ๐ (ศูนย์) ๑ (หนึ่ง) ๒ (สอง) ๓ (สาม) ๔ (สี่) ๕ (ห้า) ๖ (หก) ๗ (เจ็ด) ๘ (แปด) จนถึง ๙ (เก้า)",
-    live=True
-)
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
+
+with gr.Blocks(js=js_func) as demo:
+    gr.Interface(
+        fn=predict,
+        inputs=gr.Sketchpad(
+            label="Draw Here",
+            brush=gr.Brush(default_size=20, default_color="#FFFFFF", colors=["#FFFFFF"]),
+            image_mode="L",
+            layers=False,
+            eraser=None,
+        ),
+        outputs=gr.Label(label="Guess"),
+        title="Thai Digit Handwritten Classification",
+        description="ทดลองวาดภาพตัวอักษรเลขไทยลงใน Sketchpad ด้านล่างเพื่อทำนายผลตัวเลข ตั้งแต่ ๐ (ศูนย์) ๑ (หนึ่ง) ๒ (สอง) ๓ (สาม) ๔ (สี่) ๕ (ห้า) ๖ (หก) ๗ (เจ็ด) ๘ (แปด) จนถึง ๙ (เก้า)",
+        live=True
+    )
 
 if __name__ == "__main__":
     demo.launch()