Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = '
|
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)
|