trl-4-dnd / app.py
vishaljoshi24's picture
Chagned LLM to gpt2
5dcd105
raw
history blame
396 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 openai-community/gpt2 --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()