Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,9 +51,9 @@ class AspectSentimentPipeline(Pipeline):
|
|
51 |
sentiment = torch.argmax(logits, dim=-1).item()
|
52 |
sentiment_label = self.aspect_sentiment_model.config.id2label[sentiment]
|
53 |
sentiment_id_to_label = {
|
54 |
-
"LABEL_0": "
|
55 |
-
"LABEL_1": "
|
56 |
-
"LABEL_2": "
|
57 |
}
|
58 |
|
59 |
return sentiment_id_to_label[sentiment_label]
|
|
|
51 |
sentiment = torch.argmax(logits, dim=-1).item()
|
52 |
sentiment_label = self.aspect_sentiment_model.config.id2label[sentiment]
|
53 |
sentiment_id_to_label = {
|
54 |
+
"LABEL_0": "olumsuz",
|
55 |
+
"LABEL_1": "nötr",
|
56 |
+
"LABEL_2": "olumlu"
|
57 |
}
|
58 |
|
59 |
return sentiment_id_to_label[sentiment_label]
|