Update README.md
Browse files
README.md
CHANGED
@@ -9,6 +9,7 @@ language:
|
|
9 |
base_model:
|
10 |
- blanchefort/rubert-base-cased-sentiment
|
11 |
pipeline_tag: text-classification
|
|
|
12 |
---
|
13 |
|
14 |
# Kostya165/rubert_emotion_slicer
|
@@ -55,7 +56,4 @@ text = "Мне очень понравилось!"
|
|
55 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
|
56 |
outputs = model(**inputs)
|
57 |
predicted_class = outputs.logits.argmax(dim=-1).item()
|
58 |
-
print(f"Предсказанный класс: {predicted_class}")
|
59 |
-
|
60 |
-
|
61 |
-
|
|
|
9 |
base_model:
|
10 |
- blanchefort/rubert-base-cased-sentiment
|
11 |
pipeline_tag: text-classification
|
12 |
+
new_version: Kostya165/rubert_tiny2_russian_emotion_sentiment
|
13 |
---
|
14 |
|
15 |
# Kostya165/rubert_emotion_slicer
|
|
|
56 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
|
57 |
outputs = model(**inputs)
|
58 |
predicted_class = outputs.logits.argmax(dim=-1).item()
|
59 |
+
print(f"Предсказанный класс: {predicted_class}")
|
|
|
|
|
|