| 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"]) |
|
|
|
|
