Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -28,6 +28,7 @@ def get_target_style_embeddings(target_texts_batch):
|
|
28 |
return mean_embeddings.float().cpu().numpy()
|
29 |
|
30 |
def get_luar_embeddings(texts_batch):
|
|
|
31 |
assert set([len(texts) for texts in texts_batch]) == 1
|
32 |
episodes = texts_batch
|
33 |
tokenized_episodes = [luar_tokenizer(episode, max_length=512, padding="longest", truncation=True, return_tensors="pt").to(device) for episode in episodes]
|
|
|
28 |
return mean_embeddings.float().cpu().numpy()
|
29 |
|
30 |
def get_luar_embeddings(texts_batch):
|
31 |
+
print("HERE", texts_batch)
|
32 |
assert set([len(texts) for texts in texts_batch]) == 1
|
33 |
episodes = texts_batch
|
34 |
tokenized_episodes = [luar_tokenizer(episode, max_length=512, padding="longest", truncation=True, return_tensors="pt").to(device) for episode in episodes]
|