Update app.py

app.py

@@ -23,14 +23,13 @@ repo: Optional[Repository] = None
 # if HF_TOKEN:
 #     repo = Repository(local_dir=DATADIR, clone_from=DATASET_REPO_URL, token=HF_TOKEN)
 
-
 def neuron_export(model_id: str, task: str) -> str:
     if not model_id:
         return f"### Invalid input 🐞 Please specify a model name, got {model_id}"
 
     try:
-        api = HfApi(token=HF_TOKEN)  # Use HF_TOKEN if available, else anonymous
-        token = HF_TOKEN  # Pass token to convert only if available
+        api = HfApi(token=HF_TOKEN)
+        token = HF_TOKEN
 
         error, commit_info = convert(api=api, model_id=model_id, task=task, token=token)
         if error != "0":
@@ -61,6 +60,24 @@ def neuron_export(model_id: str, task: str) -> str:
     except Exception as e:
         return f"#### Error: {e}"
 
+# --- Custom CSS for dark/light mode fix ---
+CUSTOM_CSS = """
+<style>
+/* Default light mode style */
+input.hub-search-input {
+    color: black !important;
+    background-color: white !important;
+}
+
+/* Dark mode overrides */
+@media (prefers-color-scheme: dark) {
+    input.hub-search-input {
+        color: white !important;
+        background-color: #1f1f1f !important;
+    }
+}
+</style>
+"""
 
 TITLE_IMAGE = """
 <div style="display: block; margin-left: auto; margin-right: auto; width: 50%;">
@@ -78,17 +95,16 @@ TITLE = """
 
 DESCRIPTION = """
 Export 🤗 Transformers models hosted on the Hugging Face Hub to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
-
 *Features:*
 - Automatically opens PR with Neuron-optimized model
 - Preserves original model weights
 - Adds proper tags to model card
-
 *Note:*
 - PR creation requires the Space owner to have a valid write token set via HF_WRITE_TOKEN
 """
 
 with gr.Blocks() as demo:
+    gr.HTML(CUSTOM_CSS)
     gr.HTML(TITLE_IMAGE)
     gr.HTML(TITLE)
 
@@ -101,6 +117,7 @@ with gr.Blocks() as demo:
                 label="Hub model ID",
                 placeholder="Search for model ID on the hub",
                 search_type="model",
+                elem_classes=["hub-search-input"]
             )
             input_task = gr.Textbox(
                 value="auto",
@@ -116,7 +133,6 @@ with gr.Blocks() as demo:
         outputs=output,
     )
 
-
 if __name__ == "__main__":
     def restart_space():
         if HF_TOKEN: