Spaces:
Sleeping
Sleeping
Update tone_engine.py
Browse files- tone_engine.py +1 -0
tone_engine.py
CHANGED
|
@@ -2,6 +2,7 @@ import torch
|
|
| 2 |
import numpy as np
|
| 3 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
| 4 |
from sentence_transformers import SentenceTransformer, util
|
|
|
|
| 5 |
|
| 6 |
class ToneEnsembler:
|
| 7 |
def __init__(self, device="cpu"):
|
|
|
|
| 2 |
import numpy as np
|
| 3 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
| 4 |
from sentence_transformers import SentenceTransformer, util
|
| 5 |
+
from vibe_mapper import get_emojis_from_emotions
|
| 6 |
|
| 7 |
class ToneEnsembler:
|
| 8 |
def __init__(self, device="cpu"):
|