Titembaye commited on
Commit
3f98c65
·
1 Parent(s): 960046a

Fix: Update MODEL_PATH to point to models directory

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
9
  import os
10
 
11
  # Configuration
12
- MODEL_PATH = "models/bert-base-uncased_adversarial_final"
13
  MAX_LENGTH = 256
14
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
 
9
  import os
10
 
11
  # Configuration
12
+ MODEL_PATH = "models"
13
  MAX_LENGTH = 256
14
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15