File size: 541 Bytes
41ff2db 1f8ec05 41ff2db 1f8ec05 41ff2db 1f8ec05 a0f6eac 1f8ec05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import streamlit as st
import os
from transformers import pipeline
# from pymongo import MongoClient
# uri = os.environ["MONGO_CONNECTION_STRING"]
# client = MongoClient(uri, tlsCertificateKeyFile = "database/cert.pem")
# db = client["testing"]
# col = db["new"]
# xyz = {"This is new code678"}
# col.insert_one(xyz)
token_classifier = pipeline(model="dslim/bert-base-NER", aggregation_strategy="simple")
if ner := st.chat_input("query "):
ans = token_classifier(ner)
with st.chat_message("User"):
st.write(ans)
|