File size: 685 Bytes
3f8e7c8 2312d16 b449aa5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# 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") |