Kevin Fink
commited on
Commit
·
3270ada
1
Parent(s):
52ba344
dev
Browse files
app.py
CHANGED
@@ -167,7 +167,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
167 |
dataset['validation'].save_to_disk(f'/data/{hub_id.strip()}_validation_dataset')
|
168 |
return 'SECOND THIRD LOADED'
|
169 |
|
170 |
-
except:
|
171 |
print(f"An error occurred: {str(e)}, TB: {traceback.format_exc()}")
|
172 |
dataset = load_dataset(dataset_name.strip())
|
173 |
train_size = len(dataset['train'])
|
|
|
167 |
dataset['validation'].save_to_disk(f'/data/{hub_id.strip()}_validation_dataset')
|
168 |
return 'SECOND THIRD LOADED'
|
169 |
|
170 |
+
except Exception as e:
|
171 |
print(f"An error occurred: {str(e)}, TB: {traceback.format_exc()}")
|
172 |
dataset = load_dataset(dataset_name.strip())
|
173 |
train_size = len(dataset['train'])
|