Instructions to use datasetsANDmodels/message-extraction with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use datasetsANDmodels/message-extraction with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("datasetsANDmodels/message-extraction") model = AutoModelForSeq2SeqLM.from_pretrained("datasetsANDmodels/message-extraction") - Notebooks
- Google Colab
- Kaggle
File size: 309 Bytes
cec24a2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from transformers import pipeline
extractor = pipeline("text2text-generation", model=".")
import pandas as pd
df = pd.read_csv("message.csv")
texts=df["message"]
for intent in texts:
label=extractor(intent)[0]["generated_text"]
if label=="":
label="No message detected"
print (label )
|