Update model.py
Browse files
model.py
CHANGED
@@ -10,7 +10,7 @@ import matplotlib.pyplot as plt
|
|
10 |
from matplotlib import animation
|
11 |
from fastdtw import fastdtw # Keep this import
|
12 |
from scipy.spatial.distance import cosine
|
13 |
-
from config import MAX_TEXT_LEN,
|
14 |
from transformers import BertModel # ✅ Import BERT
|
15 |
|
16 |
tokenizer = BertTokenizer.from_pretrained("indobenchmark/indobert-base-p2")
|
|
|
10 |
from matplotlib import animation
|
11 |
from fastdtw import fastdtw # Keep this import
|
12 |
from scipy.spatial.distance import cosine
|
13 |
+
from config import MAX_TEXT_LEN, TARGET_NUM_FRAMES, BATCH_SIZE, TEACHER_FORCING_RATIO, SMOOTHING_ENABLED
|
14 |
from transformers import BertModel # ✅ Import BERT
|
15 |
|
16 |
tokenizer = BertTokenizer.from_pretrained("indobenchmark/indobert-base-p2")
|