kidwaiaun commited on
Commit
c18e9a0
·
verified ·
1 Parent(s): aea612b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # Import necessary libraries
2
  import gradio as gr
3
- from datasets import load_dataset
4
  import pandas as pd
5
  import numpy as np
6
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
 
1
  # Import necessary libraries
2
  import gradio as gr
3
+ from datasets import load_dataset, Dataset # Explicitly import Dataset class
4
  import pandas as pd
5
  import numpy as np
6
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments