| import streamlit as st | |
| from utils.uploadAndExample import add_upload | |
| from utils.config import model_dict | |
| from utils.vulnerability_classifier import label_dict | |
| with st.sidebar: | |
| # upload and example doc | |
| choice = st.sidebar.radio(label = 'Select the Document', | |
| help = 'You can upload the document \ | |
| or else you can try a example document', | |
| options = ('Upload Document', 'Try Example'), | |
| horizontal = True) | |
| add_upload(choice) | |
| # Create a list of options for the dropdown | |
| model_options = ['Llama3.1-8B','Llama3.1-70B','Llama3.1-405B','Zephyr 7B β','Mistral-7B','Mixtral-8x7B'] | |
| # Dropdown selectbox: model | |
| model_sel = st.selectbox('Select a model:', model_options) | |
| model_sel_name = model_dict[model_sel] | |
| st.session_state['model_sel_name'] = model_sel_name | |
| with st.container(): | |
| st.markdown("<h2 style='text-align: center;'> Vulnerability Analysis 3.1 </h2>", unsafe_allow_html=True) | |
| st.write(' ') |