Update app.py

app.py

@@ -50,6 +50,7 @@ def plot_curve(kn, kd):
 
 
 def compute(N, D):
+    print(N, D)
     C = to_flops(N * Bn, D * Bn)
     N_opt = n_opt(C)
     D_opt = d_opt(C)
@@ -59,7 +60,7 @@ def compute(N, D):
 
     plot_curve(kn, kd)
 
-    text = f"""Compute budget (TFLOPs): {C:.2E}\nTraining compute overhead (%): {100*compute_overhead(kn, compute_kd(kn)):.2f}\nInference cost fraction (%): {kn*100:.2f}"""
+    text = f"""Compute budget (TFLOPs): {C:.2E}\n\nTraining compute overhead (%): {100*compute_overhead(kn, kd):.2f}\n\nInference cost fraction (%): {kn*100:.2f}"""
     return text
 
 with gr.Blocks() as demo: