Spaces:
Running
Running
ThanaritKanjanametawat
commited on
Commit
•
081c69d
1
Parent(s):
1bb143a
Move everything to CPU
Browse files- ModelDriver.py +1 -2
- app.py +1 -1
ModelDriver.py
CHANGED
@@ -5,8 +5,7 @@ import torch.nn.functional as F
|
|
5 |
from torch.utils.data import TensorDataset, DataLoader
|
6 |
|
7 |
|
8 |
-
device = torch.device("
|
9 |
-
# device = torch.device("cpu")
|
10 |
class MLP(nn.Module):
|
11 |
def __init__(self, input_dim):
|
12 |
super(MLP, self).__init__()
|
|
|
5 |
from torch.utils.data import TensorDataset, DataLoader
|
6 |
|
7 |
|
8 |
+
device = torch.device("cpu")
|
|
|
9 |
class MLP(nn.Module):
|
10 |
def __init__(self, input_dim):
|
11 |
super(MLP, self).__init__()
|
app.py
CHANGED
@@ -10,7 +10,7 @@ st.write("This is a demo for GPT detection. You can use this demo to test the mo
|
|
10 |
# Add 4 options for 4 models
|
11 |
ModelOption = st.sidebar.selectbox(
|
12 |
'Which Model do you want to use?',
|
13 |
-
('RobertaSentinel'
|
14 |
)
|
15 |
|
16 |
DatasetOption = st.sidebar.selectbox(
|
|
|
10 |
# Add 4 options for 4 models
|
11 |
ModelOption = st.sidebar.selectbox(
|
12 |
'Which Model do you want to use?',
|
13 |
+
('RobertaSentinel'),
|
14 |
)
|
15 |
|
16 |
DatasetOption = st.sidebar.selectbox(
|