Spaces:

fnavales
/

hate-speech

Runtime error

fnavales commited on Oct 17, 2022

Commit

4d031cb

•

1 Parent(s): b5889be

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ def predict(model, tokenizer, sentence):
 model = ToxicCommentTagger.load_from_checkpoint(
-    '/content/drive/MyDrive/checkpoints/best-checkpoint.ckpt',
     n_classes=len(LABEL_COLUMNS)
 )

 model = ToxicCommentTagger.load_from_checkpoint(
+    './best-checkpoint.ckpt',
     n_classes=len(LABEL_COLUMNS)
 )