gregH commited on
Commit
3b0a13e
·
1 Parent(s): 0625949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch.nn.functional as F
5
  import numpy as np
6
  import random
7
 
8
- model_id = "TrustSafeAI/RADAR-Vicuna-7B"
9
  device = "cpu"
10
  detector = transformers.AutoModelForSequenceClassification.from_pretrained(model_id)
11
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
 
5
  import numpy as np
6
  import random
7
 
8
+ model_id = "gregH/Radar-Dolly-V2-3B"
9
  device = "cpu"
10
  detector = transformers.AutoModelForSequenceClassification.from_pretrained(model_id)
11
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)