File size: 681 Bytes
840e37c 7a679b4 a9706bf 840e37c a9706bf 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 |
import gradio as gr
import subprocess
def run_training():
try:
result = subprocess.run(["python", "train_abuse_model.py"], capture_output=True, text=True)
print("STDOUT:\n", result.stdout)
print("STDERR:\n", result.stderr)
return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
except Exception as 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()
|