Spaces:
Sleeping
Sleeping
add config
Browse files
config.py
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
base_model = 'mrcaelumn/yelp_restaurant_review_sentiment_analysis'
|
| 3 |
+
senti_map = {'LABEL_2': 'Positive', 'LABEL_0': 'Negative', 'LABEL_1': 'Neutral'}
|
| 4 |
+
|
| 5 |
+
absa_model = ("tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-aspect",
|
| 6 |
+
"tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-polarity")
|
| 7 |
+
|
| 8 |
+
senti_color = {'positive': 'aqua', 'neutral': 'gray', 'negative': 'orange'}
|