zerogpu for training

app.py

@@ -386,6 +386,7 @@ def make_dataset(
             )
     return dataset
 
+@spaces.GPU
 def train_model_persona(positive_text, negative_text):
     positive_list = positive_text.split('\n')
     negative_list = negative_text.split('\n')
@@ -775,11 +776,11 @@ with gr.Blocks(
     with gr.Tab(
         label="Train"
     ):
-        gr.Markdown("# 🚅 Train a new control vector")
+        gr.Markdown("# 💪 Train a new control vector")
         with gr.Row():
             with gr.Column():
                 gr.Markdown("## Persona Method")
-                gr.Markdown("Fill in the blank with three synonyms of the persona on newlines, and then three antonyms \"Act as if you are an extremely (persona) person\"")
+                gr.Markdown("Fill in the blank with three synonyms of the persona on newlines, and then three antonyms that work in this sentence, \"Act as if you are an extremely (persona) person\"")
                 persona_input_positive = gr.Text(
                     lines=3,
                     label="Positive",
@@ -796,7 +797,7 @@ with gr.Blocks(
 
             with gr.Column():
                 gr.Markdown("## Facts method")
-                gr.Markdown("Fill in the blank with a persona and its opposite within, \"Pretend to be a (persona) making statements about the world.\"")
+                gr.Markdown("Fill in the blank with a persona and its opposite that work within this sentence, \"Pretend to be a (persona) making statements about the world.\"")
                 facts_input_positive = gr.Text(
                     label="Positive",
                     placeholder="time traveler from the future")