trl-4-dnd / app.py
vishaljoshi24's picture
Amended quickstart.py for better memory allocation
4a4b568
raw
history blame contribute delete
389 Bytes
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()