import os
import gradio as gr

api_key = os.getenv("HF_TOKEN")
def start_training():
    os.system("python quickstart.py --model_name_or_path Qwen/Qwen3-0.6B --train_data_dir trl-lib/Capybara --api_key HF_TOKEN" )
    return "Evaluation completed!"

iface = gr.Interface(
    fn=start_training,
    inputs=[],
    outputs="text",
    live=True,
    allow_flagging='never'
)

iface.launch()