Update app.py

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import AutoModelForSequenceClassification, DistilBertTokenizerFast, pipeline
+
+MODEL_ID = "Krish623/sentiment-model"
+tokenizer = DistilBertTokenizerFast.from_pretrained(MODEL_ID)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None)
+
+def predict(text):
+    results = classifier(text)
+    scores = results[0] if isinstance(results[0], list) else results
+    best = max(scores, key=lambda x: x["score"])
+    return {"label": best["label"], "score": best["score"]}
+
+demo = gr.Interface(fn=predict, inputs="text", outputs="json", title="Sentiment Analysis")
+demo.launch()