Spaces:

sbstagiare
/

test

Sleeping

sbstagiare commited on Mar 11

Commit

d79632e

verified ·

1 Parent(s): be54a79

Update trainer

Files changed (1) hide show

trainer CHANGED Viewed

@@ -54,7 +54,7 @@ trainer = SFTTrainer(
 # Interface Gradio
 def train():
     trainer.train()
-    model.push_to_hub("your_hf_username/fine-tuned-model")
     return "Fine-tuning terminé et modèle uploadé sur Hugging Face !"
 gr.Interface(fn=train, inputs=[], outputs="text").launch()

 # Interface Gradio
 def train():
     trainer.train()
+    model.push_to_hub("sbstagiare/fine-tuned-model")
     return "Fine-tuning terminé et modèle uploadé sur Hugging Face !"
 gr.Interface(fn=train, inputs=[], outputs="text").launch()