Update app.py

app.py

@@ -27,6 +27,7 @@ def get_target_style_embeddings(target_texts_batch):
     mean_embeddings = torch.sum(padded_embeddings * mask, dim=1) / mask.sum(dim=1)
     return mean_embeddings.float().cpu().numpy()
 
+@torch.no_grad()
 def get_luar_embeddings(texts_batch):
     assert len(set([len(texts) for texts in texts_batch])) == 1
     episodes = texts_batch