Update app.py

app.py

@@ -126,9 +126,9 @@ def custom_strategy(prompt, flow):
 def compare_strategies(model, tokenizer, prm_model, prompt,  num_samples=5):
     print("Running comparison...")
     strategies = {
-        "Majority Voting": majority_voting(model, tokenizer, prompt, num_samples=5),
-        "Best-of-N": best_of_n(model, tokenizer, prm_model, prompt, num_samples=5),
-        "Beam Search": beam_search(model, tokenizer, prompt, num_beams=5)
+        "Majority Voting": majority_voting(model, tokenizer, prompt, num_samples),
+        "Best-of-N": best_of_n(model, tokenizer, prm_model, prompt, num_samples),
+        "Beam Search": beam_search(model, tokenizer, prompt, 5) #num_beams
         #...
     }