Enhancements ~ Optimize images, Enhance Speech2Text Plugin, Support Custom CSS in Chainlit

playground/UI/.chainlit/config.toml

@@ -57,7 +57,7 @@ hide_cot = false
 
 # Specify a CSS file that can be used to customize the user interface.
 # The CSS file can be served from the public directory or via an external link.
-custom_css = "/public/style_v1.css"
+custom_css = "path/to/your/custom.css"
 
 # Override default MUI light theme. (Check theme.ts)
 [UI.theme.light]

project/plugins/speech2text.py

@@ -13,13 +13,14 @@
 class Speech2Text(Plugin):
     model = None
 
-    def _init(self) -> None:
-        self.model = whisper.load_model("base")
+    def _init(self, model_size: str = "base") -> None:
+        self.model = whisper.load_model(model_size)
         self.device = self.model.device
 
-    def __call__(self, audio_path):
-        if self.model is None:
-            self._init()
+    def __call__(self, audio_path, model_size: str = "base"):
+        if self.model is None or self.model_size != model_size:
+            self._init(model_size)
+            self.model_size = model_size
         # load audio and pad/trim it to fit 30 seconds
         audio = whisper.load_audio(audio_path)
         audio = whisper.pad_or_trim(audio)