Update FRIDA/model.py

FRIDA/model.py

@@ -82,10 +82,10 @@ def infer(model: FridaClassifier, tokenizer: AutoTokenizer, texts: List[str], de
 labels = {0: 'non-toxic', 1: 'toxic'}
 
 
-print('loading model and tokenizer...')
-chkp_dir = './'    # CHANGE ON YOUR DIR WITH HEAD WEIGHTS!
-model, tokenizer = load_model(os.path.join(chkp_dir, "classifier_head.pth"))
-print('loaded.')
+#print('loading model and tokenizer...')
+#chkp_dir = './'    # CHANGE ON YOUR DIR WITH HEAD WEIGHTS!
+#model, tokenizer = load_model(os.path.join(chkp_dir, "classifier_head.pth"))
+#print('loaded.')
 
 
 from typing import List