created app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+import subprocess
+
+def run_training():
+    try:
+        # Run train.py using subprocess and capture output
+        result = subprocess.run(["python", "train.py"], capture_output=True, text=True)
+        # Return stdout if success, otherwise stderr
+        return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
+    except Exception as e:
+        return f"Exception occurred:\n{str(e)}"
+
+# Define a simple Gradio interface with one button
+demo = gr.Interface(
+    fn=run_training,
+    inputs=[],
+    outputs="text",
+    title="Run Model Training",
+    description="Click the button to execute train.py. This will use GPU if available."
+)
+
+demo.launch()