Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -191,7 +191,7 @@ def main():
|
|
| 191 |
|
| 192 |
# Load Dataset
|
| 193 |
train_dataset = load_dataset(data_source, tokenizer, uploaded_file=uploaded_file)
|
| 194 |
-
|
| 195 |
if data_source == "demo":
|
| 196 |
data = ["Sample text data for model training. This can be replaced with actual data for better performance."]
|
| 197 |
elif uploaded_file is not None:
|
|
|
|
| 191 |
|
| 192 |
# Load Dataset
|
| 193 |
train_dataset = load_dataset(data_source, tokenizer, uploaded_file=uploaded_file)
|
| 194 |
+
def load_dataset(data_source="demo", tokenizer=None, uploaded_file=None):
|
| 195 |
if data_source == "demo":
|
| 196 |
data = ["Sample text data for model training. This can be replaced with actual data for better performance."]
|
| 197 |
elif uploaded_file is not None:
|