File size: 546 Bytes
840e37c 2032430 840e37c 2032430 9364099 2032430 9364099 2032430 a9706bf 2032430 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
from train_abuse_model import run_training
with gr.Blocks() as demo:
gr.Markdown("## 🧠 Abuse Detection Fine-Tuning App")
gr.Markdown(
"⚠️ **Important:** Keep this tab open and prevent your computer from sleeping while training runs."
)
with gr.Row():
start_btn = gr.Button("🚀 Start Training")
output_box = gr.Textbox(label="Live Training Logs", lines=25, interactive=False)
start_btn.click(fn=run_training, outputs=output_box)
if __name__ == "__main__":
demo.launch()
|