trl-4-dnd / app.py
vishaljoshi24's picture
Changed LLM back to Qwen
f91cad5
raw
history blame
392 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/Qwen2.5-0.5B --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()