Kevin Fink
commited on
Commit
·
f916b82
1
Parent(s):
73a715d
dev
Browse files
app.py
CHANGED
|
@@ -8,6 +8,8 @@ import traceback
|
|
| 8 |
|
| 9 |
import os
|
| 10 |
from huggingface_hub import login
|
|
|
|
|
|
|
| 11 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|
| 12 |
try:
|
| 13 |
login(api_key.strip())
|
|
|
|
| 8 |
|
| 9 |
import os
|
| 10 |
from huggingface_hub import login
|
| 11 |
+
|
| 12 |
+
@spaces.GPU
|
| 13 |
def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch_size, lr, grad):
|
| 14 |
try:
|
| 15 |
login(api_key.strip())
|