Upload app.py

app.py

@@ -59,19 +59,23 @@ MODEL_REPO = "meettilavat/imagecaptioning"
 SUBFOLDER_PREFIX = "outputs/blip2_full_ft_stage2"
 LOCAL_DIR = Path(os.environ["HF_HOME"]) / "models" / "imagecaptioning"
 DEFAULT_PROMPT = "Describe the image in detail."
-SPINNER_CSS = """
+SPINNER_MARKUP = """
+<div class="caption-spinner">
+    <div class="caption-spinner__loader" aria-hidden="true"></div>
+    <span role="status">Generating caption...</span>
+</div>
 <style>
-#caption-spinner {
+.caption-spinner {
     display: flex;
     align-items: center;
     gap: 0.5rem;
     font-size: 0.95rem;
 }
-#caption-spinner .caption-spinner__loader {
+.caption-spinner__loader {
     width: 20px;
     height: 20px;
-    border: 3px solid var(--neutral-400, rgba(0, 0, 0, 0.25));
-    border-top-color: var(--body-text-color, rgba(0, 0, 0, 0.75));
+    border: 3px solid rgba(0, 0, 0, 0.25);
+    border-top-color: rgba(0, 0, 0, 0.75);
     border-radius: 50%;
     animation: caption-spin 0.75s linear infinite;
 }
@@ -81,7 +85,7 @@ SPINNER_CSS = """
     }
 }
 </style>
-"""
+""".strip()
 
 
 def _allow_patterns() -> Iterable[str]:
@@ -260,7 +264,6 @@ with gr.Blocks(title="BLIP-2 Image Captioning") as demo:
     gr.Markdown(
         "Upload an image, tweak decoding settings, and optionally compare beam widths side by side."
     )
-    gr.HTML(SPINNER_CSS)
 
     with gr.Row():
         with gr.Column(scale=6, min_width=320):
@@ -311,12 +314,10 @@ with gr.Blocks(title="BLIP-2 Image Captioning") as demo:
                 "Compare beams to balance diversity vs. precision."
             )
             spinner_display = gr.HTML(
-                value=(
-                    '<div class="caption-spinner__loader" aria-hidden="true"></div>'
-                    "<span role=\"status\">Generating caption...</span>"
-                ),
+                value=SPINNER_MARKUP,
                 visible=False,
-                elem_id="caption-spinner",
+                show_label=False,
+                height=48,
             )
 
     beam_mode_input.change(