Spaces:

Thanarit
/

GPT-Detection-Demo

Running

ThanaritKanjanametawat commited on Sep 26, 2023

Commit

1fb6014

1 Parent(s): dd9c120

Move everything to CPU 2

Files changed (1) hide show

ModelDriver.py CHANGED Viewed

@@ -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=torch.device("cpu"))
     model = model.to(torch.device('cpu'))
     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=torch.device("cpu"))
     model = model.to(torch.device('cpu'))
     model.eval()

     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=torch.device("cpu")))
     model = model.to(torch.device('cpu'))
     model.eval()
     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=torch.device("cpu")))
     model = model.to(torch.device('cpu'))
     model.eval()