msmaje commited on
Commit
4b7e5cc
·
verified ·
1 Parent(s): 7603677

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- dataset_name,
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,