Add application file

app.py

@@ -11,28 +11,33 @@ import gradio as gr
 from utils import approach1_isoflops, approach2_derivative, approach3_isoloss
 
 
-def compute_optimal_vocab(Nnv: float,
-                        flops: float, 
-                        ):
-    
-    if flops is None:
+def compute_optimal_vocab(Nnv, flops):
+    try:
+        Nnv = float(Nnv)
+    except ValueError:
+        return "Invalid input for Non-vocabulary Parameters."
+
+    if flops:
+        try:
+            flops = float(flops)
+        except ValueError:
+            return "Invalid input for FLOPs."
+
+    if flops is None or flops == "":
         Vopt_app1 = approach1_isoflops(Nnv)
         Vopt_app2 = approach2_derivative(Nnv)
-        Vopt_app3 = approach3_isoloss(Nnv) 
+        Vopt_app3 = approach3_isoloss(Nnv)
     else:
-        Vopt_app1,  Vopt_app2 = None, None
-        Vopt_app3 = approach3_isoloss(Nnv, flops) 
-    # Vopt_app1, Vopt_app2, Vopt_app3=1.,2.,3.
-    
-    results = 'test'
-    # results = f"The optimal vocabulary size is:\nApproach 1: {Vopt_app1}\nApproach 2: {Vopt_app2}\nApproach 3: {Vopt_app3}"
-    return results
+        Vopt_app1, Vopt_app2 = None, None
+        Vopt_app3 = approach3_isoloss(Nnv, flops)
 
+    results = f"The optimal vocabulary size is:\nApproach 1: {Vopt_app1}\nApproach 2: {Vopt_app2}\nApproach 3: {Vopt_app3}"
+    return results
 
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown(
-            """<img src="https://github.com/sail-sg/scaling-with-vocab/blob/main/figures/figure_1_left_scaling_v5.png" style="float: left;" width="250" height="250"><h1>The Optimal Vocabulary Size Predictor</h1>
+            """<h1>The Optimal Vocabulary Size Predictor</h1>
             This tool is used to predict the optimal vocabulary size given the non-vocabulary parameters.
             We provide 3 ways for prediction:
 
@@ -47,8 +52,8 @@ with gr.Blocks() as demo:
             """)
         
         with gr.Row():
-            Nnv = gr.Textbox(label="Non-vocabulary Parameters", value=7*10**9)
-            flops = gr.Textbox(label="FLOPs", placeholder="Optional (e.g. 7.05*10**21)")
+            Nnv = gr.Textbox(label="Non-vocabulary Parameters", value=str(7*10**9))
+            flops = gr.Textbox(label="FLOPs", placeholder="Optional (e.g. 7.05e21)")
             output_text = gr.Textbox(label="Prediction")
         with gr.Row():
             btn = gr.Button("Compute the optimal vocabulary size")