Update app.py

app.py

@@ -55,10 +55,17 @@ def run_tinystyler_batch(source_texts, target_texts_batch, reranking, temperatur
         do_sample=True, 
         temperature=temperature, 
         top_p=top_p,
-        max_length=1024
+        max_length=1024,
+        n=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)]
+
+    # Get best based on re-ranking
+    generated_texts = [texts[0] for texts in generated_texts]
+    
     return generated_texts
     
 def run_tinystyler(source_text, target_texts, reranking, temperature, top_p):