Bux fix when running on CPU

app.py

@@ -398,13 +398,16 @@ with gr.Blocks(
     
 
     # Header
-    gr.Markdown("# 🧠 LLM Mind Control")
+    if cuda:
+        gr.Markdown("# 🧠 LLM Mind Control")
+    else:
+        gr.Markdown(""" 🧠 LLM Mind Control
+
+*Warning: running on CPU will be very slow*""")
     gr.Markdown("""Unlike prompting, direct weight manipulation lets you fine-tune the amount of a personality
 trait or topic. Enabled through [Representation Engineering](https://arxiv.org/abs/2310.01405)
 via the [repeng](https://pypi.org/project/repeng) library.
 [Watch a demo](https://youtu.be/gYZPGVafD7M) for usage tips.""")
-    if not cuda:
-        gr.Markdown("*Warning: running on CPU will be very slow*")
 
     with gr.Row():
         # Left Column: Control Vectors and advanced settings