Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,8 @@ 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
|
12 |
detector.eval()
|
13 |
detector.to(device)
|
14 |
|
|
|
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)
|
12 |
detector.eval()
|
13 |
detector.to(device)
|
14 |
|