Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def fine_tune_model(base_model_name, dataset_name):
|
|
| 35 |
print("### Train example")
|
| 36 |
print(dataset["train"][:1])
|
| 37 |
print("### Test example")
|
| 38 |
-
print(dataset["test"][:1)
|
| 39 |
print("###")
|
| 40 |
|
| 41 |
# Load model
|
|
@@ -60,7 +60,7 @@ def fine_tune_model(base_model_name, dataset_name):
|
|
| 60 |
print("### Train example")
|
| 61 |
print(dataset["train"][:1])
|
| 62 |
print("### Test example")
|
| 63 |
-
print(dataset["test"][:1)
|
| 64 |
print("###")
|
| 65 |
|
| 66 |
# Split dataset into training and validation sets
|
|
|
|
| 35 |
print("### Train example")
|
| 36 |
print(dataset["train"][:1])
|
| 37 |
print("### Test example")
|
| 38 |
+
print(dataset["test"][:1])
|
| 39 |
print("###")
|
| 40 |
|
| 41 |
# Load model
|
|
|
|
| 60 |
print("### Train example")
|
| 61 |
print(dataset["train"][:1])
|
| 62 |
print("### Test example")
|
| 63 |
+
print(dataset["test"][:1])
|
| 64 |
print("###")
|
| 65 |
|
| 66 |
# Split dataset into training and validation sets
|