x

app.py

@@ -13,13 +13,13 @@ from utils import approach1_isoflops, approach2_derivative, approach3_isoloss
 
 def compute_optimal_vocab(Nnv, flops):
     try:
-        Nnv = float(Nnv)
+        Nnv = float(eval(Nnv))
     except ValueError:
         return "Invalid input for Non-vocabulary Parameters."
 
     if flops:
         try:
-            flops = float(flops)
+            flops = float(eval(flops))
         except ValueError:
             return "Invalid input for FLOPs."
 
@@ -37,7 +37,7 @@ def compute_optimal_vocab(Nnv, flops):
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown(
-            """<h1>The Optimal Vocabulary Size Predictor</h1>
+            """<img src="figures/figure_1_left_scaling_v5.png" style="float: left;" width="250" height="250"><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: