Update README.md

README.md

@@ -46,13 +46,13 @@ $ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: applica
 Or Python API:
 
 ```
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
-
-model = AutoModelForSequenceClassification.from_pretrained("crodri/MassiveCatalanIntents", use_auth_token=True)
+from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("crodri/MassiveCatalanIntents", use_auth_token=True)
 
-inputs = tokenizer("I love AutoTrain", return_tensors="pt")
+model = AutoModelForSequenceClassification.from_pretrained("crodri/MassiveCatalanIntents", use_auth_token=True)
+
+pipe = pipeline("text-classification",model=model,tokenizer=tokenizer)
 
-outputs = model(**inputs)
+result = pipe("afegeix a la llista de la compra un litre de llet")
 ```