Add requirements

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
+id2label = {0: "Negative", 1: "Positive"}
+label2id = {"Negative":0, "Positive":1}
+
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained('distilbert/distilbert-base-uncased', add_prefix_space=True)
-model = AutoModelForSequenceClassification.from_pretrained('AmirRghp/distilbert-base-uncasedimdb-text-classification')
+model = AutoModelForSequenceClassification.from_pretrained(
+    model="AmirRghp/distilbert-base-uncasedimdb-text-classification", num_labels=2, id2label=id2label, label2id=label2id)
 
 # Create the pipeline
 classifier = pipeline('text-classification', model=model, tokenizer=tokenizer)