Update app.py

app.py

@@ -192,6 +192,15 @@ def create_interface():
                     value=check_gpu_status(),
                     interactive=False,
                 )
+
+                # hidden constant input for model name
+                model_name = gr.Textbox(
+                    value="Qwen/Qwen2.5-0.5B",
+                    label="Base Model",
+                    interactive=False,
+                    visible=False
+                )
+
                 num_epochs = gr.Slider(1, 3, value=1, step=1, label="Epochs")
                 batch_size = gr.Slider(1, 4, value=2, step=1, label="Batch Size")
                 learning_rate = gr.Number(value=5e-5, label="Learning Rate")
@@ -205,15 +214,14 @@ def create_interface():
                     value="Click 'Start Fine-tuning' to train on the Gita dataset and upload to your model repo.",
                 )
 
+        # ✅ Fixed: pass components, not strings
         train_btn.click(
             fn=train_model,
-            inputs=["Qwen/Qwen2.5-0.5B", num_epochs, batch_size, learning_rate],
+            inputs=[model_name, num_epochs, batch_size, learning_rate],
             outputs=output,
         )
 
     return demo
-
-
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()