Update app.py

app.py

@@ -71,6 +71,7 @@ def fine_tune_model(base_model_id, dataset):
         save_on_each_node=True,
         load_best_model_at_end=True,
         eval_strategy="steps",
+        push_to_hub=True,
     )
     
     # Create Trainer instance