Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ def get_target_style_embeddings(target_texts_batch):
|
|
| 23 |
mean_embeddings = torch.sum(padded_embeddings * mask, dim=1) / mask.sum(dim=1)
|
| 24 |
return mean_embeddings.cpu().numpy()
|
| 25 |
|
| 26 |
-
def run_tinystyler_batch(source_texts,
|
| 27 |
inputs = tokenizer(source_texts, return_tensors="pt")
|
| 28 |
target_style_embeddings = get_target_style_embeddings(target_texts_batch)
|
| 29 |
|
|
|
|
| 23 |
mean_embeddings = torch.sum(padded_embeddings * mask, dim=1) / mask.sum(dim=1)
|
| 24 |
return mean_embeddings.cpu().numpy()
|
| 25 |
|
| 26 |
+
def run_tinystyler_batch(source_texts, target_texts_batch, reranking, temperature, top_p):
|
| 27 |
inputs = tokenizer(source_texts, return_tensors="pt")
|
| 28 |
target_style_embeddings = get_target_style_embeddings(target_texts_batch)
|
| 29 |
|