gregH commited on
Commit
2bd10be
·
1 Parent(s): 89617e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ 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()
 
7
 
8
  model_id = "TrustSafeAI/RADAR-Vicuna-7B"
9
  device = "cpu"
10
+
11
  detector = transformers.AutoModelForSequenceClassification.from_pretrained(model_id)
12
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_id)
13
  detector.eval()