Kevin Fink
commited on
Commit
·
37472a5
1
Parent(s):
172d00c
init
Browse files
app.py
CHANGED
|
@@ -4,7 +4,6 @@ from datasets import load_dataset
|
|
| 4 |
import traceback
|
| 5 |
|
| 6 |
def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr, grad):
|
| 7 |
-
'''
|
| 8 |
try:
|
| 9 |
|
| 10 |
# Load the dataset
|
|
@@ -53,7 +52,6 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
|
|
| 53 |
trainer.push_to_hub(commit_message="Training complete!")
|
| 54 |
except Exception as e:
|
| 55 |
return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
|
| 56 |
-
'''
|
| 57 |
return 'DONE!'#model
|
| 58 |
'''
|
| 59 |
# Define Gradio interface
|
|
|
|
| 4 |
import traceback
|
| 5 |
|
| 6 |
def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr, grad):
|
|
|
|
| 7 |
try:
|
| 8 |
|
| 9 |
# Load the dataset
|
|
|
|
| 52 |
trainer.push_to_hub(commit_message="Training complete!")
|
| 53 |
except Exception as e:
|
| 54 |
return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
|
|
|
|
| 55 |
return 'DONE!'#model
|
| 56 |
'''
|
| 57 |
# Define Gradio interface
|