Update app.py

app.py

@@ -1,6 +1,52 @@
 import gradio as gr
 from PIL import Image
 from inference.main import MultiModalPhi2
+from __future__ import annotations
+from typing import Iterable
+import gradio as gr
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
+import time
+
+
+class Seafoam(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.emerald,
+        secondary_hue: colors.Color | str = colors.blue,
+        neutral_hue: colors.Color | str = colors.gray,
+        spacing_size: sizes.Size | str = sizes.spacing_md,
+        radius_size: sizes.Size | str = sizes.radius_md,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Quicksand"),
+            "ui-sans-serif",
+            "sans-serif",
+        ),
+        font_mono: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"),
+            "ui-monospace",
+            "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            spacing_size=spacing_size,
+            radius_size=radius_size,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+
+
+seafoam = Seafoam()
 
 messages = []
 
@@ -60,18 +106,15 @@ def run(history, text, image, audio_upload, audio_mic):
     return history, None, None, None, None
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=seafoam) as demo:
     gr.Markdown("## MulitModal Phi2 Model Pretraining and Finetuning from Scratch")
 
-# with gr.Column(scale=8):
-# with gr.Box():
     with gr.Row():
         chatbot = gr.Chatbot(
             avatar_images=("🧑", "🤖"),
             height=450,
         )
 
-    # with gr.Box():
     with gr.Row():
         # Adding a Textbox with a placeholder "write prompt"
         prompt = gr.Textbox(
@@ -88,7 +131,7 @@ with gr.Blocks() as demo:
         )
     with gr.Row():
         # Adding a Button
-        submit = gr.Button()
+        submit = gr.Button(value = "Submit")
         clear = gr.Button(value="Clear")
 
     submit.click(