Spaces:

Thanarit
/

GPT-Detection-Demo

Running

ThanaritKanjanametawat commited on Sep 26, 2023

Commit

cf62d27

1 Parent(s): 57bafce

change the device to cpu only

Files changed (1) hide show

ModelDriver.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import RobertaTokenizer, RobertaForSequenceClassification, Rob
 import torch
 import torch.nn as nn
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 class MLP(nn.Module):
     def __init__(self, input_dim):
         super(MLP, self).__init__()

 import torch
 import torch.nn as nn
+device = torch.device("cpu")
 class MLP(nn.Module):
     def __init__(self, input_dim):
         super(MLP, self).__init__()