hantupocong commited on
Commit
b5736a5
·
verified ·
1 Parent(s): e55790d

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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, TARGET_NUM_FRAME, BATCH_SIZE, TEACHER_FORCING_RATIO, SMOOTHING_ENABLED
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")