Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
-# Manually load tokenizer from base model
+
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = AutoModelForSequenceClassification.from_pretrained("King-8/interview_statements")
 
@@ -9,9 +9,10 @@ classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
 
 def classify_statement(statement):
     prediction = classifier(statement)[0]
+    
     label = prediction["label"]
-    score = round(prediction["score"] * 100, 2)
-    return f"Category: {label} (Confidence: {score}%)"
+    
+    return f"Predicted Category: {label}"
 
 gr.Interface(
     fn=classify_statement,