Update app.py

app.py

@@ -77,12 +77,12 @@ class TextWindowProcessor:
 
 class TextClassifier:
     def __init__(self):
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        if self.device.type == 'cpu':
-            # Enable CPU optimizations
+        # Set thread configuration before any model loading or parallel work
+        if not torch.cuda.is_available():
             torch.set_num_threads(MAX_WORKERS)
             torch.set_num_interop_threads(MAX_WORKERS)
             
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.model_name = MODEL_NAME
         self.tokenizer = None
         self.model = None
@@ -105,11 +105,6 @@ class TextClassifier:
             self.model_name,
             num_labels=2
         ).to(self.device)
-        
-        if self.device.type == 'cpu':
-            # Enable faster CPU performance without TorchScript
-            torch.set_num_threads(MAX_WORKERS)
-            torch.set_num_interop_threads(MAX_WORKERS)
             
         model_path = "model_20250209_184929_acc1.0000.pt"
         if os.path.exists(model_path):