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 )