update

demo/binary_classifier_demo.py

@@ -7,12 +7,6 @@ import os
 from model_utils import load_model, classify_text
 from binoculars_utils import initialize_binoculars, compute_scores
 
-# Initialize Binoculars models
-bino_chat, bino_coder = initialize_binoculars()
-
-# Load binary classifier model
-model, scaler, label_encoder, imputer = load_model()
-
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 MINIMUM_TOKENS = 200
 
@@ -54,6 +48,12 @@ def run_binary_classifier(text, show_analysis=False):
     if len(text.strip()) < MINIMUM_TOKENS:
         return gr.Markdown(f"Текст слишком короткий. Требуется минимум {MINIMUM_TOKENS} символов."), None, None
     
+    # Initialize Binoculars models
+    bino_chat, bino_coder = initialize_binoculars()
+
+    # Load binary classifier model
+    model, scaler, label_encoder, imputer = load_model()
+
     # Compute scores using binoculars
     scores = compute_scores(text, bino_chat, bino_coder)