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 |
|