Update app.py

app.py

@@ -12,6 +12,11 @@ from nltk.stem import WordNetLemmatizer
 import re
 import string
 from huggingface_hub import hf_hub_download
+import warnings
+from sklearn.exceptions import InconsistentVersionWarning
+
+# Suppress scikit-learn warning
+warnings.filterwarnings("ignore", category=InconsistentVersionWarning)
 
 # Download NLTK resources
 nltk.download('punkt', quiet=True)
@@ -21,7 +26,7 @@ nltk.download('omw-1.4', quiet=True)
 # Initialize lemmatizer
 lemmatizer = WordNetLemmatizer()
 
-# Define LuongAttention (placeholder for loading)
+# Define LuongAttention
 class LuongAttention(tf.keras.layers.Layer):
     def __init__(self, **kwargs):
         super(LuongAttention, self).__init__(**kwargs)
@@ -122,9 +127,6 @@ h1 { color: #1a73e8; text-align: center; }
 .footer { text-align: center; color: #666; }
 .prob-bar { margin-top: 10px; }
 button { border-radius: 6px; }
-.dark { background: #1e1e1e; color: #ffffff; }
-.dark h1 { color: #4a90e2; }
-.dark .footer { color: #aaa; }
 """
 
 # Gradio interface
@@ -164,7 +166,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     
     # Theme toggle logic
     def toggle_theme():
-        return gr.themes.Dark() if demo.theme.name == "soft" else gr.themes.Soft()
+        return {"theme": gr.themes.Dark()} if demo.theme.name == "soft" else {"theme": gr.themes.Soft()}
     
     # Bind functions
     predict_btn.click(
@@ -174,8 +176,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     )
     theme_toggle.click(
         fn=toggle_theme,
-        outputs=None,
-        _js="() => { document.body.classList.toggle('dark'); }"
+        inputs=None,
+        outputs=[demo]
     )
     
     gr.Markdown(