| import gradio as gr | |
| import subprocess | |
| def run_training(): | |
| # Execute train.py and capture its output | |
| result = subprocess.run(["python", "train.py"], capture_output=True, text=True) | |
| return result.stdout + "\n" + result.stderr | |
| # Create a Gradio interface with no inputs and a text output for logs | |
| iface = gr.Interface( | |
| fn=run_training, | |
| inputs=[], | |
| outputs="text", | |
| title="LLaMA LoRA Fine-Tuning", | |
| description="Click the button below to start fine-tuning the LLaMA 3.2 3B Instruct model using PEFT/LoRA." | |
| ) | |
| iface.launch() | |