Update app.py

app.py

@@ -2,11 +2,16 @@ import gradio as gr
 import pandas as pd
 from transformers import BertTokenizer, BertForSequenceClassification
 import torch
+from datasets import load_dataset
 
 # Load pre-trained TinyBERT model and tokenizer
 tokenizer = BertTokenizer.from_pretrained('huawei-noah/TinyBERT_General_4L_312D')
 model = BertForSequenceClassification.from_pretrained('huawei-noah/TinyBERT_General_4L_312D')
 
+# Load dataset from Hugging Face repository
+# Replace 'your-username' and 'your-dataset-name' with actual values
+dataset = load_dataset('https://huggingface.co/SharmaAmit1818', data_files='data.csv')
+
 # Function to process the CSV file and generate predictions
 def process_csv(file):
     try: