Move everything to CPU

ModelDriver.py

@@ -62,7 +62,7 @@ def RobertaClassifierOpenGPTInference(input_text):
     tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
     model_path = "ClassifierCheckpoint/RobertaClassifierOpenGPT.pth"
     model = RobertaForSequenceClassification.from_pretrained('roberta-base', num_labels=2)
-    model.load_state_dict(torch.load(model_path, map_location=device))
+    model.load_state_dict(torch.load(model_path, map_location=device), strict=False)
     model = model.to(device)
     model.eval()
 
@@ -84,7 +84,7 @@ def RobertaClassifierCSAbstractInference(input_text):
     tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
     model_path = "ClassifierCheckpoint/RobertaClassifierCSAbstract.pth"
     model = RobertaForSequenceClassification.from_pretrained('roberta-base', num_labels=2)
-    model.load_state_dict(torch.load(model_path, map_location=device))
+    model.load_state_dict(torch.load(model_path, map_location=device), strict=False)
     model = model.to(device)
     model.eval()
 

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'),
+    ('RobertaSentinel', 'RobertaClassifier'),
 )
 
 DatasetOption = st.sidebar.selectbox(