Update app.py

app.py

@@ -111,7 +111,7 @@ def calc_flops(vocab_size, hidden_size, sequence_length, num_layers, kv_size_rat
 
 
 # ---- Gradio Interface ---- #
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme="ysharma/TransformerCalculatorNew") as demo:
     with gr.Accordion("Credits and General Idea", open=False):
         gr.Markdown("""
         This app is a re-creation of [this calculator](https://github.com/EleutherAI/cookbook/tree/main/calc) from EleutherAI.