update

demo/binary_classifier_demo.py

@@ -7,7 +7,7 @@ import os
 from model_utils import load_model, classify_text
 from binoculars_utils import initialize_binoculars, compute_scores
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+DEVICE = torch.device('cuda')
 MINIMUM_TOKENS = 200
 
 SAMPLE_TEXT = """Привет! Я хотел бы рассказать вам о своём опыте путешествия по Петербургу. Невероятный город с богатой историей и красивой архитектурой. Особенно запомнился Эрмитаж с его огромной коллекцией произведений искусства. Также понравилась прогулка по каналам города, где можно увидеть множество старинных мостов и зданий."""
@@ -44,6 +44,7 @@ css = """
 }
 """
 
+@spaces.GPU
 def run_binary_classifier(text, show_analysis=False):
     if len(text.strip()) < MINIMUM_TOKENS:
         return gr.Markdown(f"Текст слишком короткий. Требуется минимум {MINIMUM_TOKENS} символов."), None, None