ZeroGPU

demo/demo.py

@@ -1,6 +1,7 @@
 __all__ = ["app"]
 
 import gradio as gr
+import spaces
 from binoculars import Binoculars
 
 BINO = Binoculars()
@@ -12,6 +13,7 @@ def count_tokens(text):
     return len(TOKENIZER(text).input_ids)
 
 
+@spaces.GPU
 def run_detector(input_str):
     if count_tokens(input_str) < MINIMUM_TOKENS:
         gr.Warning(f"Too short length. Need minimum {MINIMUM_TOKENS} tokens to run Binoculars.")