Move everything to CPU

ModelDriver.py

@@ -5,8 +5,7 @@ import torch.nn.functional as F
 from torch.utils.data import TensorDataset, DataLoader
 
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-# device = torch.device("cpu")
+device = torch.device("cpu")
 class MLP(nn.Module):
     def __init__(self, input_dim):
         super(MLP, self).__init__()

app.py

@@ -10,7 +10,7 @@ st.write("This is a demo for GPT detection. You can use this demo to test the mo
 # Add 4 options for 4 models
 ModelOption = st.sidebar.selectbox(
     'Which Model do you want to use?',
-    ('RobertaSentinel', 'RobertaClassifier'),
+    ('RobertaSentinel'),
 )
 
 DatasetOption = st.sidebar.selectbox(