Adapted demo to new theme

app.py

@@ -2,7 +2,7 @@ import os
 from dotenv import load_dotenv
 import gradio as gr
 from gradio.components import Textbox, Button, Slider, Checkbox
-from AinaTheme import AinaGradioTheme
+from AinaTheme import theme
 from sagemaker_endpoint import invoke_endpoint
 
 load_dotenv()
@@ -16,7 +16,6 @@ def submit_input(input_, max_new_tokens, repetition_penalty, top_k, top_p, do_sa
     if input_.strip() == "":  
         gr.Warning('Not possible to inference an empty input')
         return None
-    
     model_parameters = {
         "max_new_tokens": max_new_tokens,
         "repetition_penalty": repetition_penalty,
@@ -53,7 +52,7 @@ def clear():
     )
     
 def gradio_app():
-    with gr.Blocks(**AinaGradioTheme().get_kwargs()) as demo:
+    with gr.Blocks(theme=theme) as demo:
         with gr.Row():
             with gr.Column(scale=0.1):
                 gr.Image("ginesta_small.jpg", elem_id="flor-banner", scale=1, height=256, width=256, show_label=False, show_download_button = False, show_share_button = False)
@@ -69,7 +68,7 @@ def gradio_app():
                  
                     """
                 )
-        with gr.Row( equal_height=False):
+        with gr.Row(equal_height=True):
             with gr.Column(variant="panel"):
                 placeholder_max_token = Textbox(
                     visible=False,