chore: moved generate button

app.py

@@ -100,6 +100,8 @@ def main():
                 v2.ui()
 
             with gr.Column():
+                generate_btn = gr.Button(value="Generate tags", variant="primary")
+
                 with gr.Group():
                     output_text = gr.TextArea(label="Output tags", interactive=False)
                     copy_btn = gr.Button(
@@ -225,7 +227,7 @@ def main():
                     ],
                 )
 
-        v2.get_generate_btn().click(
+        generate_btn.click(
             parse_upsampling_output(v2.on_generate),
             inputs=[
                 *v2.get_inputs(),

v2.py

@@ -210,8 +210,6 @@ class V2UI:
             value=list(ALL_MODELS.keys())[0],
         )
 
-        self.generate_btn = gr.Button(value="Generate", variant="primary")
-
         self.input_components = [
             model_name,
             input_copyright,
@@ -224,8 +222,5 @@ class V2UI:
             input_ban_tags,
         ]
 
-    def get_generate_btn(self) -> gr.Button:
-        return self.generate_btn
-
     def get_inputs(self) -> list[Component]:
         return self.input_components