Update app.py

app.py

@@ -4,6 +4,12 @@ import numpy as np
 from PIL import Image
 import cv2
 from torchvision import models,transforms
+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
 core = Core()
 
 # Read model to OpenVINO Runtime
@@ -83,6 +89,44 @@ items = {
 
 
 }
+class Davinci_Eye(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.stone,
+        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("IBM Plex Mono"),
+            "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,
+        )
+
+
+davincieye = Davinci_Eye()
 
 colormap={v:[i for i in k] for k,v in color_map.items()}
 
@@ -122,6 +166,6 @@ demo=gr.Interface(fn=segment_image,inputs=gr.Image(type='filepath'),
                             "R005_ch1_video_03_00-19-10-11.jpg",
                             "R006_ch1_video_01_00-45-02-10.jpg",
                             "R013_ch1_video_03_00-40-17-11.jpg"],
-                  #themes=gr.themes.Glass(primary_hue=gr.themes.colors.blue,secondary_hue=gr.themes.colors.blue),
+                  theme=davincieye.set(loader_color='#65aab1'),
                   title="Davinci Eye(Quantized for CPU)")
 demo.launch()