Spaces:
Runtime error
Runtime error
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]
|