Update app.py

app.py

@@ -373,7 +373,7 @@ def train_model_old(epochs):
 def gradio_train(epochs):
     # Gradio training interface function
     model, training_log = train_model(int(epochs))
-    to_hub(model)
+    #to_hub(model)
     return f"{training_log}\n\nModel trained for {epochs} epochs and pushed to {model_repo_id}"
 
 def gradio_inference(input_image):