Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,10 @@ def fine_tune_model(base_model_name, dataset_name):
|
|
| 32 |
|
| 33 |
print("### Dataset")
|
| 34 |
print(dataset)
|
| 35 |
-
print(
|
|
|
|
|
|
|
|
|
|
| 36 |
print("###")
|
| 37 |
|
| 38 |
# Load model
|
|
@@ -54,7 +57,10 @@ def fine_tune_model(base_model_name, dataset_name):
|
|
| 54 |
|
| 55 |
print("### Pre-processed dataset")
|
| 56 |
print(dataset)
|
| 57 |
-
print(
|
|
|
|
|
|
|
|
|
|
| 58 |
print("###")
|
| 59 |
|
| 60 |
# Split dataset into training and validation sets
|
|
|
|
| 32 |
|
| 33 |
print("### Dataset")
|
| 34 |
print(dataset)
|
| 35 |
+
print("###")
|
| 36 |
+
print(dataset["train"][:5])
|
| 37 |
+
print("###")
|
| 38 |
+
print(dataset["test"][:5])
|
| 39 |
print("###")
|
| 40 |
|
| 41 |
# Load model
|
|
|
|
| 57 |
|
| 58 |
print("### Pre-processed dataset")
|
| 59 |
print(dataset)
|
| 60 |
+
print("###")
|
| 61 |
+
print(dataset["train"][:5])
|
| 62 |
+
print("###")
|
| 63 |
+
print(dataset["test"][:5])
|
| 64 |
print("###")
|
| 65 |
|
| 66 |
# Split dataset into training and validation sets
|