Add application file

app.py

@@ -32,7 +32,7 @@ def compute_optimal_vocab(Nnv: float,
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown(
-            """<img src="figure_1_left_scaling_v5.png" style="float: left;" width="250" height="250"><h1>The Optimal Vocabulari Size Predictor</h1>
+            """<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>
             This tool is used to predict the optimal vocabulary size given the non-vocabulary parameters.
             We provide 3 ways for prediction:
 
@@ -53,11 +53,11 @@ with gr.Blocks() as demo:
         with gr.Row():
             btn = gr.Button("Compute the optimal vocabulary size")
 
-          btn.click(
-                compute_optimal_vocab,
-                inputs=[Nnv, flops],
-                outputs=output_text
-            )
+        btn.click(
+            compute_optimal_vocab,
+            inputs=[Nnv, flops],
+            outputs=output_text
+        )
 demo.launch()
 
 # import gradio as gr