File size: 535 Bytes
505405b c13bc85 505405b f7a043e 505405b |
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
from pymongo import MongoClient
import os
from transformers import pipeline
# uri = os.environ["MONGO_CONNECTION_STRING"]
# client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem")
# db = client["testing"]
# col = db["new"]
qna = pipeline("question-answering")
knowledge = "My name is Ankush and I am excited to know about Artificial Intelligence"
if query := st.chat_input("Question: "):
ans = qna(question=query, context=knowledge)
with st.chat_message("User"):
st.write(ans)
|