Update app.py

app.py

@@ -1,14 +1,26 @@
+import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
+# Load model
 model_name = "shayeedahmed/psyche-bert-emotion-classifier"
-
-# Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
+nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+# Define inference function
+def classify_text(text):
+    result = nlp(text)
+    return result
 
-# Create a pipeline
-classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+# Gradio interface
+iface = gr.Interface(
+    fn=classify_text,
+    inputs=gr.Textbox(label="Enter text here"),
+    outputs=gr.JSON(label="Prediction"),
+    title="Emotion Classifier",
+    description="Classifies text into emotions"
+)
 
-# Test
-result = classifier("Hello, how are you?")
-print(result)
+# Launch the app (this is mandatory!)
+if __name__ == "__main__":
+    iface.launch()