Update app.py

app.py

@@ -23,7 +23,7 @@ def n_opt(C):
 def d_opt(C):
     return (1/G) * ((C/6) ** (alpha / (alpha+beta)))
 
-def get_kd(kn):
+def compute_kd(kn):
     frac = (A/B)*(G**(-alpha-beta))    
     kd = (1-((kn**-alpha -1)*frac))**(1/(-beta))
     return kd
@@ -55,10 +55,11 @@ def compute(N, D):
     D_opt = d_opt(C)
 
     kn = N/N_opt
+    kd = compute_kd(kn)
 
-    plot_curve(kn, 100*overhead(kn, get_kd(kn)))
+    plot_curve(kn, kd)
 
-    text = f"""Compute budget (TFLOPs): {C:.2E}\nTraining compute overhead (%): {100*compute_overhead(kn, get_kd(kn)):.2f}\nInference cost fraction (%): {kn*100:.2f}"""
+    text = f"""Compute budget (TFLOPs): {C:.2E}\nTraining compute overhead (%): {100*compute_overhead(kn, compute_kd(kn)):.2f}\nInference cost fraction (%): {kn*100:.2f}"""
     return text
 
 with gr.Blocks() as demo: