Update finetune.py

finetune.py

@@ -168,8 +168,25 @@ old_state_dict = model.state_dict
 model.state_dict = (
     lambda self, *_, **__: get_peft_model_state_dict(self, old_state_dict())
 ).__get__(model, type(model))
-print("Training")
-trainer.train()
+
+import gradio as gr
+
+def train():
+    # Call your trainer's train() function here
+    trainer.train()
+    print("Training complete.")  # optional message to display when training is done
+
+iface = gr.Interface(
+    fn=train,
+    inputs=None,
+    outputs=None,
+    title="Training Interface",
+    description="Click the button to start training.",
+    theme="default",
+    layout="vertical",
+    allow_flagging=False,
+    allow_screenshot=False
+)
 
 model.save_pretrained(OUTPUT_DIR)
 repo.push_to_hub(OUTPUT_DIR, commit_message="Ft model")