| # set path | |
| import glob, os, sys; | |
| sys.path.append('../utils') | |
| #import needed libraries | |
| import seaborn as sns | |
| import matplotlib.pyplot as plt | |
| import numpy as np | |
| import pandas as pd | |
| import streamlit as st | |
| from utils.target_classifier import load_targetClassifier, target_classification | |
| from utils.config import get_classifier_params | |
| # Declare all the necessary variables | |
| classifier_identifier = 'target' | |
| params = get_classifier_params(classifier_identifier) | |
| ## Labels dictionary ### | |
| _lab_dict = { | |
| 'NEGATIVE':'NO TARGET INFO', | |
| 'TARGET':'TARGET', | |
| } | |
| # Download model from HF Hub | |
| vg_model = SetFitModel.from_pretrained("leavoigt/vulnerable_groups") |