Update README.md

README.md

@@ -29,8 +29,20 @@ result summary:
 usage:
 
 ```
-model = AutoModelForSequenceClassification.from_pretrained("premsa/political-bias-prediction-allsides-BERT")
-tokenizer = AutoTokenizer.from_pretrained("premsa/political-bias-prediction-allsides-BERT")
-nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
-print(nlp("the masses are controlled by media."))
+from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
+
+
+def main(repository: str):
+
+    model = AutoModelForSequenceClassification.from_pretrained(repository)
+
+    tokenizer = AutoTokenizer.from_pretrained(repository)
+
+    nlp = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+    print(nlp("the masses are controlled by media. I am left wing"))
+
+if __name__ == "__main__":
+    main(repository="premsa/political-bias-prediction-allsides-BERT")
+
 ```