Upload folder using huggingface_hub

app.py

@@ -574,25 +574,29 @@ with gr.Blocks() as demo:
     with gr.Tab("🏋️ Train"):
         gr.Markdown("Train the μ-Net live on this hardware.")
         
-        with gr.Row():
-            task = gr.Dropdown(
-                ["Coherence Prediction", "Sequence Prediction"],
-                value="Coherence Prediction",
-                label="Task"
-            )
-            epochs = gr.Slider(50, 500, value=200, step=10, label="Epochs")
-        
-        with gr.Row():
-            lr = gr.Number(value=0.001, label="Learning Rate")
-            lambda_c = gr.Number(value=0.01, label="λ (coherence regularization)")
+        task = gr.Radio(
+            ["Coherence Prediction", "Sequence Prediction"],
+            value="Coherence Prediction",
+            label="Task"
+        )
+        epochs = gr.Slider(50, 500, value=100, step=10, label="Epochs")
+        lr = gr.Number(value=0.001, label="Learning Rate")
+        lambda_c = gr.Number(value=0.01, label="λ coherence")
         
         train_btn = gr.Button("🚀 Train μ-Net", variant="primary")
         
-        train_log = gr.Markdown(label="Training Log")
-        train_curve = gr.Markdown(label="Training Curve")
+        train_log = gr.Textbox(label="Training Log", lines=20, interactive=False)
+        train_curve = gr.Textbox(label="Training Curve", lines=15, interactive=False)
+        
+        def safe_train(task, epochs, lr, lam):
+            try:
+                return train_model(task, epochs, lr, lam)
+            except Exception as e:
+                import traceback
+                return f"ERROR: {e}\n\n{traceback.format_exc()}", ""
         
         train_btn.click(
-            train_model,
+            safe_train,
             inputs=[task, epochs, lr, lambda_c],
             outputs=[train_log, train_curve]
         )
@@ -605,7 +609,7 @@ with gr.Blocks() as demo:
             label="8 ratio values (comma-separated)"
         )
         infer_btn = gr.Button("Predict", variant="primary")
-        infer_output = gr.Markdown()
+        infer_output = gr.Textbox(label="Result", lines=15, interactive=False)
         infer_btn.click(run_inference, inputs=input_box, outputs=infer_output)
     
     with gr.Tab("📤 Push to Hub"):
@@ -616,7 +620,7 @@ with gr.Blocks() as demo:
             label="Repository ID"
         )
         push_btn = gr.Button("Push Model", variant="primary")
-        push_output = gr.Markdown()
+        push_output = gr.Textbox(label="Status", lines=3, interactive=False)
         push_btn.click(push_to_hub, inputs=repo_input, outputs=push_output)
     
     with gr.Tab("🧠 Architecture"):

mu_net_trained.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9a1e606de65304e8e2f90cd9d1fc3baeee139b88cdb4c7dad80c84002124c84
+size 157633