app.py

app.py

@@ -33,7 +33,10 @@ def encode_domain(domain: str, max_len: int = 64):
 # Load model from HuggingFace Hub
 MODEL_NAME = "ccss17/dga-transformer-encoder"  # Replace with your model
 print(f"Loading model from {MODEL_NAME}...")
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+model = AutoModelForSequenceClassification.from_pretrained(
+    MODEL_NAME,
+    trust_remote_code=True,  # Required for custom model architecture
+)
 model.eval()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")