requirements

app.py

@@ -2,14 +2,13 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
-model_name = "rafaelcarvalhoj/emotion-classifier"
+model_name = "Bittar/NLP_Emotions"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 mapping = {
-    0: 'anger',
-    1: 'joy',
-    2: 'fear'
+    0: 'negative',
+    1: 'positive'
 }
 
 def predict(text):

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch