Update app.py

app.py

@@ -122,8 +122,6 @@ body { font-family: 'Inter', sans-serif; }
 /* Ensure label text inside columns is readable */
 #prediction-label .label-name { font-weight: bold; font-size: 1.1em; }
 #prediction-label .confidence { font-size: 1em; }
-/* Theme should make label text light, but force if needed: */
-/* #prediction-label { color: #e5e7eb; } */
 
 
 /* Footer styling */
@@ -145,17 +143,9 @@ body { font-family: 'Inter', sans-serif; }
 """
 
 # --- Gradio Interface using Blocks and Theme ---
-# Apply .dark() to the theme
-theme = gr.themes.Soft(
-    primary_hue="emerald",
-    secondary_hue="blue",
-    neutral_hue="slate",
-    radius_size=gr.themes.sizes.radius_lg,
-    spacing_size=gr.themes.sizes.spacing_lg,
-).dark() # <<< APPLY DARK MODE HERE
-
-
-with gr.Blocks(theme=theme, css=css) as iface:
+# Use the theme string identifier for the dark mode variant
+# Other options: "default/dark", "monochrome/dark", "glass/dark"
+with gr.Blocks(theme="soft/dark", css=css) as iface:  # <<< CHANGE IS HERE
     # Title and Description
     gr.Markdown("# AI vs Human Image Detector", elem_id="app-title")
     gr.Markdown(
@@ -165,7 +155,7 @@ with gr.Blocks(theme=theme, css=css) as iface:
     )
 
     # Main layout
-    with gr.Row(variant='panel'): # Panel might look different in dark theme, adjust if needed
+    with gr.Row(variant='panel'):
         with gr.Column(scale=1, min_width=300, elem_id="input-column"):
             image_input = gr.Image(
                 type="pil",
@@ -181,7 +171,6 @@ with gr.Blocks(theme=theme, css=css) as iface:
                 num_top_classes=2,
                 label="Classification",
                 elem_id="prediction-label"
-                # The theme should now correctly style the label text for dark mode
             )
 
     # Examples Section
@@ -193,18 +182,19 @@ with gr.Blocks(theme=theme, css=css) as iface:
             fn=classify_image,
             cache_examples=True,
             label="✨ Click an Example to Try!"
-            # Examples appearance will also adapt to the dark theme
         )
 
     # Footer / Article section
-    # Removed explicit model ID formatting from Markdown string, use f-string
     gr.Markdown(f"""
        ---
+       **How it Works:**
        This application uses a fine-tuned [SigLIP](https://huggingface.co/docs/transformers/model_doc/siglip) vision model
        specifically trained to differentiate between images generated by Artificial Intelligence and those created by humans.
 
-       You can find the model card here: <a href='https://huggingface.co/{MODEL_IDENTIFIER}' target='_blank'>{MODEL_IDENTIFIER}</a>
+       **Model:**
+       *   You can find the model card here: <a href='https://huggingface.co/{MODEL_IDENTIFIER}' target='_blank'>{MODEL_IDENTIFIER}</a>
 
+       **Training Code:**
        Fine tuning code available at [https://exnrt.com/blog/ai/fine-tuning-siglip2/](https://exnrt.com/blog/ai/fine-tuning-siglip2/).
        """,
        elem_id="app-footer"