| import gradio as gr | |
| from train_abuse_model import run_training, evaluate_saved_model, push_model_to_hub | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## 🧠 Abuse Detection Fine-Tuning App") | |
| gr.Markdown("⚠️ Keep this tab open while training or evaluating.") | |
| with gr.Row(): | |
| start_btn = gr.Button("🚀 Start Training") | |
| eval_btn = gr.Button("🔍 Evaluate Trained Model") | |
| push_btn = gr.Button("📤 Push Model to Hub") | |
| output_box = gr.Textbox(label="Logs", lines=25, interactive=False) | |
| start_btn.click(fn=run_training, outputs=output_box) | |
| eval_btn.click(fn=evaluate_saved_model, outputs=output_box) | |
| push_btn.click(fn=push_model_to_hub, outputs=output_box) | |
| if __name__ == "__main__": | |
| demo.launch() | |