Upload app.py with huggingface_hub

app.py

@@ -36,7 +36,7 @@ USER_TEXT = (
 processor = AutoProcessor.from_pretrained(MODEL_ID)
 model = AutoModelForImageTextToText.from_pretrained(
     MODEL_ID,
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
 ).to("cuda")
 
 
@@ -96,7 +96,7 @@ CSS = """
 .dark .gradio-container { color: var(--body-text-color); }
 """
 
-with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
+with gr.Blocks() as demo:
     gr.Markdown(
         "# 🖥️ WebBrain VL 2 450M — Browser Screenshot Understanding\n"
         "Upload a browser screenshot to get a structured six-section observation "
@@ -135,7 +135,6 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
                     label="Structured Observation",
                     lines=24,
                     max_lines=50,
-                    show_copy_button=True,
                 )
 
         run_btn.click(
@@ -160,4 +159,4 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS) as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True, theme=gr.themes.Citrus(), css=CSS)