File size: 754 Bytes
840e37c 54299e5 840e37c 2032430 9364099 54299e5 2032430 54299e5 9364099 54299e5 9364099 2032430 54299e5 a9706bf 2032430 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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()
|