feat: demo app

app.py

@@ -105,6 +105,7 @@ def strip_edge_punct(word: str):
     return leading, word, trailing
 
 
+@spaces.GPU
 @torch.inference_mode()
 def _predict(text: str, top_k: int = 5):
     mdl = model.to("cuda")
@@ -131,7 +132,6 @@ def _predict(text: str, top_k: int = 5):
     return (filtered or raw)[:top_k]
 
 
-@spaces.GPU
 def on_word_click(text: str, evt: gr.SelectData):
     if evt.value is None or not str(evt.value).strip():
         return gr.update(), gr.update()
@@ -185,7 +185,7 @@ CSS = """
 """
 
 
-with gr.Blocks(title="ModernBERTić Large", css=CSS) as demo:
+with gr.Blocks(title="ModernBERTić Large") as demo:
     gr.Markdown(
         """
         # ModernBERTić Large
@@ -242,4 +242,4 @@ with gr.Blocks(title="ModernBERTić Large", css=CSS) as demo:
 
 
 if __name__ == "__main__":
-    demo.queue().launch(ssr_mode=False)
+    demo.queue().launch(ssr_mode=False, css=CSS)