Spaces:
Running
Running
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 |
|