Update app.py

app.py

@@ -65,9 +65,9 @@ duo_complete_pass_at_k = duo_df.groupby('Model')[['Runs', 'Successes']].apply(la
 }, index=['pass@1'])).reset_index()
 
 complete_pass_at_k = df.groupby('Model')[['Runs', 'Successes']].apply(lambda x: pd.Series({
-    'pass@1': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 1).mean(),
-    'pass@5': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 5).mean(),
-    'pass@10': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 10).mean()
+    'pass@1': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 1).mean()
+    #'pass@5': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 5).mean(),
+    #'pass@10': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 10).mean()
 }, index=['pass@1', 'pass@5', 'pass@10'])).reset_index()
     
 with gr.Blocks() as demo: