File size: 801 Bytes
840e37c a3bf050 7a679b4 a3bf050 840e37c a3bf050 840e37c a9706bf a3bf050 840e37c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import gradio as gr
import subprocess
def run_training():
try:
print("✅ Training started.")
result = subprocess.run(["python", "train_abuse_model.py"], capture_output=True, text=True)
print("✅ Training finished.")
print("STDOUT:", result.stdout)
print("STDERR:", result.stderr)
return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
except Exception as e:
print("❌ Exception occurred:", e)
return f"Exception occurred:\n{str(e)}"
# Define a simple Gradio interface with one button
demo = gr.Interface(
fn=run_training,
inputs=[],
outputs="text",
title="Run Model Training",
description="Click the button to execute train.py. This will use GPU if available."
)
demo.launch()
|