PirateXX commited on
Commit
5021878
1 Parent(s): c699581

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
16
  # model = RobertaForSequenceClassification.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN, config = config)
17
 
18
  device = 'cuda' if cuda.is_available() else 'cpu'
19
- tokenizer = AutoTokenizer.from_pretrained("PirateXX/AI-Content-Detector", use_auth_token= ACCESS_TOKEN)
20
- model = AutoModelForSequenceClassification.from_pretrained("PirateXX/AI-Content-Detector", use_auth_token= ACCESS_TOKEN)
21
  model.to(device)
22
 
23
  # model_name = "roberta-base"
 
16
  # model = RobertaForSequenceClassification.from_pretrained("PirateXX/ChatGPT-Text-Detector", use_auth_token= ACCESS_TOKEN, config = config)
17
 
18
  device = 'cuda' if cuda.is_available() else 'cpu'
19
+ tokenizer = AutoTokenizer.from_pretrained("PirateXX/AI-Content-Detector", token= ACCESS_TOKEN)
20
+ model = AutoModelForSequenceClassification.from_pretrained("PirateXX/AI-Content-Detector", token= ACCESS_TOKEN)
21
  model.to(device)
22
 
23
  # model_name = "roberta-base"