complaint_mgnt / app.py
agshubhi's picture
Create app.py
b650cce verified
raw
history blame contribute delete
295 Bytes
from bertopic import BERTopic
import streamlit as st
from transformers import pipeline
import pandas as pd
docs=pd.read_csv('/rows.csv')
topic_model = BERTopic()
model = topic_model.fit(docs)
text=st.text_area('enter customer feedback')
if text:
out =model.transform(text)
st.json(out)