PirateXX commited on
Commit
9123c31
·
1 Parent(s): e8af4f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model = RobertaForSequenceClassification.from_pretrained("PirateXX/ChatGPT_Detec
13
  model_name = "roberta-base"
14
  tokenizer = RobertaTokenizer.from_pretrained(model_name, map_location=torch.device('cpu'))
15
 
16
- device = 'cuda' if cuda.is_available() else 'cpu'
17
  model_id = "gpt2"
18
  modelgpt2 = GPT2LMHeadModel.from_pretrained(model_id).to(device)
19
  tokenizergpt2 = GPT2TokenizerFast.from_pretrained(model_id)
 
13
  model_name = "roberta-base"
14
  tokenizer = RobertaTokenizer.from_pretrained(model_name, map_location=torch.device('cpu'))
15
 
16
+ device = 'cpu'
17
  model_id = "gpt2"
18
  modelgpt2 = GPT2LMHeadModel.from_pretrained(model_id).to(device)
19
  tokenizergpt2 = GPT2TokenizerFast.from_pretrained(model_id)