PirateXX commited on
Commit
a240a54
·
1 Parent(s): bcfb3b3

Update app.py

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