Create app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from trainer import run_finetune
+
+
+def start_training(base_model, dataset_file, epochs):
+    output = run_finetune(
+        base_model=base_model,
+        dataset_path=dataset_file.name,
+        epochs=int(epochs),
+    )
+    return output
+
+
+ui = gr.Interface(
+    fn=start_training,
+    inputs=[
+        gr.Textbox(value="distilbert-base-uncased", label="Base model"),
+        gr.File(label="Dataset (jsonl)"),
+        gr.Number(value=3, label="Epochs")
+    ],
+    outputs=gr.Textbox(),
+    title="HuggingFace Fine-Tuning Space"
+)
+
+ui.launch(server_name="0.0.0.0", server_port=7860)