Update app.py

app.py

@@ -56,12 +56,13 @@ def run_tinystyler_batch(source_texts, target_texts_batch, reranking, temperatur
         temperature=temperature, 
         top_p=top_p,
         max_length=1024,
-        n=reranking,
+        num_return_sequences=reranking,
     )
     
     generated_texts = tokenizer.batch_decode(output, skip_special_tokens=True)
 
-    output_luar_embeddings = [get_luar_embeddings([[candidates[i]] for candidates in generated_texts]) for i in range(reranking)]
+    # Evaluate candidates
+    candidates_luar_embeddings = [get_luar_embeddings([[candidates[i]] for candidates in generated_texts]) for i in range(reranking)]
 
     # Get best based on re-ranking
     generated_texts = [texts[0] for texts in generated_texts]