Spaces:

fserfati13
/

chatgpt_detector

Runtime error

fserfati13 commited on May 13

Commit

a6b1837

•

1 Parent(s): 32ba95a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from text_preprocessing import preprocess_text
 # Load the fine-tuned BERT model
 model = RobertaClass()
-model.load_state_dict(torch.load('model.bin',
                                  map_location=torch.device('cpu')))
 # Load the tokenizer

 # Load the fine-tuned BERT model
 model = RobertaClass()
+model.load_state_dict(torch.load('trained_model.bin',
                                  map_location=torch.device('cpu')))
 # Load the tokenizer