message-extraction / usage.py
datasetsANDmodels's picture
Upload 2 files
cec24a2
raw
history blame contribute delete
309 Bytes
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 )