gradio fix

app.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 import traceback
 
 def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr, grad):
+    '''
     try:
         
         # Load the dataset
@@ -51,7 +52,8 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
         trainer.train()
         trainer.push_to_hub(commit_message="Training complete!")
     except Exception as e:
-        return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"    
+        return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
+    '''
     return 'DONE!'#model
 '''
 # Define Gradio interface