Create app.py

app.py

@@ -0,0 +1,19 @@
+import gradio as gr
+import os
+
+# Function to start training
+def start_training():
+    os.system("python train.py")  # Runs your training script
+    return "Training Started! 🚀 Check logs in 'train_log.txt'."
+
+# Create Gradio UI
+iface = gr.Interface(
+    fn=start_training,
+    inputs=[], 
+    outputs="text",
+    title="Train TinyLlama on Hugging Face Spaces",
+    description="Click the button below to start fine-tuning TinyLlama."
+)
+
+# Launch the Gradio app
+iface.launch()