| 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["something"] | |
| qna = pipeline("question-answering") | |
| with open("knowledge.txt", "r") as file: | |
| data = file.read() | |
| knowledge = data | |
| if query := st.chat_input("Question: "): | |
| ans = qna(question=query, context=knowledge) | |
| with st.chat_message("User"): | |
| st.write(ans["answer"]) | |