Update app.py

changed the output to give only the predicted string class.

app.py

@@ -30,7 +30,7 @@ def predict(text):
         predicted_class_id = predictions.argmax().item()
         confidence = predictions.max().item()
         category_name = categories[predicted_class_id]
-    return f"Category: {category_name}\nConfidence: {confidence:.4f}"
+    return category_name
 
 # Create Gradio interface
 iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="DistilBERT Text Classifier")