Spaces:

jishnuprakash
/

lex_glue_ecthrA

Build error

jishnuprakash commited on Sep 23, 2022

Commit

44340f9

1 Parent(s): a46f9af

path updated

Files changed (1) hide show

home.py CHANGED Viewed

@@ -43,9 +43,7 @@ expander.write("""This is minimal user interface implemetation to view and inter
 #Load trained model
 @st.cache(allow_output_mutation=True)
 def load_model():
-    trained_model = ut.LexGlueTagger.load_from_checkpoint(
-                        os.path.join(os.getcwd(), ut.checkpoint_dir, ut.check_filename+'.ckpt'),
-                        num_classes = ut.num_classes)
     #Initialise BERT tokenizer
     tokenizer = AutoTokenizer.from_pretrained(ut.bert_model)
     #Set to Eval and freeze to avoid weight update

 #Load trained model
 @st.cache(allow_output_mutation=True)
 def load_model():
+    trained_model = ut.LexGlueTagger.load_from_checkpoint(ut.check_filename+'.ckpt', num_classes = ut.num_classes)
     #Initialise BERT tokenizer
     tokenizer = AutoTokenizer.from_pretrained(ut.bert_model)
     #Set to Eval and freeze to avoid weight update