AjayP13 commited on
Commit
fc5a09a
·
verified ·
1 Parent(s): 8ea3dab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, target_example_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
 
 
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