app / app.py
waliiid's picture
Update app.py
e9e791f
raw
history blame contribute delete
908 Bytes
import streamlit as st
import time
import pinecone
from langchain.vectorstores import Pinecone
from langchain.chains import RetrievalQA
from transformers import (
BitsAndBytesConfig,
AutoConfig,
AutoModelForCausalLM,
AutoTokenizer,
pipeline,
)
import os, tempfile
import torch
from torch import bfloat16
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.document_loaders import PyPDFLoader
from langchain.llms import HuggingFacePipeline
from streamlit_chat import message
from streamlit_option_menu import option_menu
import json
import streamlit.components.v1 as components
with st.sidebar:
sidebar_selection = option_menu(" Menu", ['Question Answering','json' ],
icons=['bi bi-patch-question-fill','bi bi-filetype-json'], menu_icon="cast", default_index=1)
sidebar_selection