Update README.md

README.md

@@ -9,6 +9,7 @@ metrics:
 base_model:
 - distilbert/distilbert-base-uncased
 pipeline_tag: text-classification
+library_name: transformers
 ---
 This repository contains a fine-tuned DistilBERT model for sentiment classification of Amazon product reviews The model classifies a given review into two classes: Positive and Negative
 
@@ -51,4 +52,4 @@ outputs = model(**inputs)
 logits = outputs.logits
 sentiment = torch.argmax(logits, dim=1).item()
 
-print(f"Predicted sentiment: {'Positive' if sentiment else 'Negative'}")
+print(f"Predicted sentiment: {'Positive' if sentiment else 'Negative'}")