trl-4-dnd / app.py
vishaljoshi24's picture
Modified app.py to correct typo in LLM name
6999e27
raw
history blame
390 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()