File size: 696 Bytes
840e37c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
import subprocess
def run_training():
try:
# Run train.py using subprocess and capture output
result = subprocess.run(["python", "train.py"], capture_output=True, text=True)
# Return stdout if success, otherwise 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()
|