Add training pipeline

app.py

@@ -191,4 +191,4 @@ with gr.Blocks(title="Rewrite — Training Space") as demo:
     refresh_btn.click(fn=get_logs, outputs=log_box)
     status_btn.click(fn=get_status, outputs=status_box)
 
-demo.launch()
+demo.launch(server_name="0.0.0.0")

requirements.txt

@@ -1,4 +1,5 @@
 audioop-lts
+jinja2<3.1.4
 transformers>=4.40.0
 peft>=0.10.0
 datasets>=2.19.0