Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -602,7 +602,7 @@ with gr.Blocks(title="BERT Complaint Classifier") as app:
|
|
| 602 |
train_btn.click(
|
| 603 |
train_model,
|
| 604 |
inputs=[
|
| 605 |
-
|
| 606 |
num_epochs,
|
| 607 |
batch_size,
|
| 608 |
learning_rate,
|
|
|
|
| 602 |
train_btn.click(
|
| 603 |
train_model,
|
| 604 |
inputs=[
|
| 605 |
+
dataset_file,
|
| 606 |
num_epochs,
|
| 607 |
batch_size,
|
| 608 |
learning_rate,
|