Update README.md

README.md

@@ -123,35 +123,22 @@ For reliable results, the classifier should be applied to documents of sufficien
 Use example:
 
 ```python
-from simpletransformers.classification import ClassificationModel
+from transformers import pipeline
 
-model_args ={
-             "num_train_epochs": 5,
-             "learning_rate": 8e-06,
-             "train_batch_size": 32,
-             "max_seq_length": 512,
-            "silent": True,
-             }   
+# Load a multi-class classification pipeline - if the model runs on CPU, comment out "device"
+classifier = pipeline("text-classification", model="classla/multilingual-IPTC-news-topic-classifier", device=0)
+
+# Example texts to classify
+texts = [
+    """Slovenian handball team makes it to Paris Olympics semifinal Lille, 8 August - Slovenia defeated Norway 33:28 in the Olympic men's handball tournament in Lille late on Wednesday to advance to the semifinal where they will face Denmark on Friday evening. This is the best result the team has so far achieved at the Olympic Games and one of the best performances in the history of Slovenia's team sports squads.""",
+	"""Moment dog sparks house fire after chewing power bank An indoor monitoring camera shows the moment a dog unintentionally caused a house fire after chewing on a portable lithium-ion battery power bank. In the video released by Tulsa Fire Department in Oklahoma, two dogs and a cat can be seen in the living room before a spark started the fire that spread within minutes. Tulsa Fire Department public information officer Andy Little said the pets escaped through a dog door, and according to local media the family was also evacuated safely. "Had there not been a dog door, they very well could have passed away," he told CBS affiliate KOTV."""]
+
+# Classify the texts
+results = classifier(texts)
 
-model = ClassificationModel(
-    "xlmroberta", "classla/multilingual-IPTC-news-topic-classifier", use_cuda=True,
-    args=model_args
-    
-)
-predictions, logit_output = model.predict([
-                                "Slovenian handball team makes it to Paris Olympics semifinal Lille, 8 August - Slovenia defeated Norway 33:28 in the Olympic men's handball tournament in Lille late on Wednesday to advance to the semifinal where they will face Denmark on Friday evening. This is the best result the team has so far achieved at the Olympic Games and one of the best performances in the history of Slovenia's team sports squads.",
-"""Moment dog sparks house fire after chewing power bank
-An indoor monitoring camera shows the moment a dog unintentionally caused a house fire after chewing on a portable lithium-ion battery power bank.
-In the video released by Tulsa Fire Department in Oklahoma, two dogs and a cat can be seen in the living room before a spark started the fire that spread within minutes.
-Tulsa Fire Department public information officer Andy Little said the pets escaped through a dog door, and according to local media the family was also evacuated safely.
-"Had there not been a dog door, they very well could have passed away," he told CBS affiliate KOTV.."""]
-                                        )
-
-predictions
-# Output: array([3, 5])
-
-[model.config.id2label[i] for i in predictions]
-# Output: ['sport', 'disaster, accident and emergency incident']
+# Output the results
+for result in results:
+    print(result)
 
 ```