Fix meta tensor error: use device_map in from_pretrained instead of .to(device)

plagiarism_detection.py

@@ -78,12 +78,15 @@ def ai_plagiarism_detection(text, threshold=0.5, show_results=False):
 
     # Model and Tokenizer Directory
     model_directory = "desklib/ai-text-detector-v1.01"
-    # Load tokenizer and model
-    tokenizer = AutoTokenizer.from_pretrained(model_directory)
-    model = DesklibAIDetectionModel.from_pretrained(model_directory)
     # Set up device
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model.to(device)
+    # Load tokenizer and model directly to the target device to avoid meta tensor issues
+    tokenizer = AutoTokenizer.from_pretrained(model_directory)
+    model = DesklibAIDetectionModel.from_pretrained(
+        model_directory,
+        device_map=device,
+        low_cpu_mem_usage=False
+    )
     # Predict
     probability, ai_detected = predict_single_text(text, model, tokenizer, device, threshold=threshold)
     # to print results