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")
|