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:
|