Spaces:

ndiy
/

restaurant-ces

Sleeping

restaurant-ces / config.py

add config

a874f18 5 months ago

No virus

398 Bytes


	base_model = 'mrcaelumn/yelp_restaurant_review_sentiment_analysis'
	senti_map = {'LABEL_2': 'Positive', 'LABEL_0': 'Negative', 'LABEL_1': 'Neutral'}

	absa_model = ("tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-aspect",
	"tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-polarity")

	senti_color = {'positive': 'aqua', 'neutral': 'gray', 'negative': 'orange'}