Update app.py

app.py

@@ -81,21 +81,31 @@ def fine_tune_model(dataset_path, dataset_file, split, colonne_input_ids, colonn
     # Lancement de l'entraînement
     trainer.train()
 
-    # Sauvegarde du modèle
-    trainer.save_model("./")
-    tokenizer.save_pretrained("./")
-
-    # Push vers Hugging Face Hub
-    api = HfApi()
-    api.upload_folder(
-        folder_path="./",
+#    # Sauvegarde du modèle
+#    trainer.save_model("./")
+#    tokenizer.save_pretrained("./")
+#
+#    # Push vers Hugging Face Hub
+#    api = HfApi()
+#    api.upload_folder(
+#        folder_path="./",
+#        repo_id=model_name,  # Utiliser le même repo_id que le modèle original
+#        repo_type="model",
+#        token=hf_token,
+#    )
+#
+#    return "Fine-tuning terminé et modèle sauvegardé."
+
+    # Sauvegarde et push vers Hugging Face Hub
+    trainer.push_to_hub(
         repo_id=model_name,  # Utiliser le même repo_id que le modèle original
-        repo_type="model",
-        token=hf_token,
+        commit_message="Fine-tuning terminé",
+        private=False,  # Définir sur True si le dépôt doit être privé
     )
 
     return "Fine-tuning terminé et modèle sauvegardé."
 
+
 # Interface Gradio
 with gr.Blocks() as demo:
     with gr.Tab("Chatbot"):