Update app.py

app.py

@@ -6,8 +6,6 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-import gradio as gr
-
        
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -87,75 +85,4 @@ agent = CodeAgent(
 )
 
 
-# GradioUI(agent).launch()
-
-
-css = """
-.message-row {
-    justify-content: space-evenly !important;
-}
-.message-bubble-border {
-    border-radius: 6px !important;
-}
-.message-buttons-bot, .message-buttons-user {
-    right: 10px !important;
-    left: auto !important;
-    bottom: 2px !important;
-}
-.dark.message-bubble-border {
-    border-color: #15172c !important;
-}
-.dark.user {
-    background: #10132c !important;
-}
-.dark.assistant.dark, .dark.pending.dark {
-    background: #020417 !important;
-}
-"""
-
-# Create Gradio interface
-gr.ChatInterface(
-    predict,
-    title=EMOJI + " " + MODEL_NAME,
-    description=DESCRIPTION,
-    examples=[
-        ["Can you solve the equation 2x + 3 = 11 for x?"],
-        ["Write an epic poem about Ancient Rome."],
-        ["Who was the first person to walk on the Moon?"],
-        ["Use a list comprehension to create a list of squares for numbers from 1 to 10."],
-        ["Recommend some popular science fiction books."],
-        ["Can you write a short story about a time-traveling detective?"]
-    ],
-    additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False),
-    additional_inputs=[
-        gr.Textbox("Perform the task to the best of your ability.", label="System prompt"),
-        gr.Slider(0, 1, 0.8, label="Temperature"),
-        gr.Slider(128, 4096, 512, label="Max new tokens"),
-        gr.Slider(1, 80, 40, label="Top K sampling"),
-        gr.Slider(0, 2, 1.1, label="Repetition penalty"),
-        gr.Slider(0, 1, 0.95, label="Top P sampling"),
-    ],
-    theme=gr.themes.Soft(primary_hue=COLOR).set(
-        background_fill_primary_dark="#020417",
-        background_fill_secondary_dark="#020417",
-        body_background_fill_dark="#020417",
-        block_background_fill_dark="#020417",
-        block_border_width="1px",
-        block_title_background_fill_dark="#15172c",
-        input_background_fill_dark="#15172c",
-        button_secondary_background_fill_dark="#15172c",
-        border_color_accent_dark="#15172c",
-        border_color_primary_dark="#15172c",
-        color_accent_soft_dark="#10132c",
-        code_background_fill_dark="#15172c",
-    ),
-    css=css,
-    retry_btn="Retry",
-    undo_btn="Undo",
-    clear_btn="Clear",
-    submit_btn="Send",
-    chatbot=gr.Chatbot(
-        scale=1,
-        show_copy_button=True
-    )
-).queue().launch()
+GradioUI(agent).launch()