Update app.py

app.py

@@ -418,13 +418,13 @@ def train_model_old(epochs):
     return model, "\n".join(output_text)
 
 def gradio_train(epochs):
-    """Gradio training interface function"""
+    # Gradio training interface function
     model, training_log = train_model(int(epochs))
     to_hub(model)
     return f"{training_log}\n\nModel trained for {epochs} epochs and pushed to {model_repo_id}"
 
 def gradio_inference(input_image):
-    """Gradio inference interface function"""
+    # Gradio inference interface function
     output_image = run_inference(input_image)  # Assuming `run_inference` returns a tuple (output_image, other_data)
     rp(output_image)
     # If `run_inference` returns a tuple, you should only return the image part