yinuozhang commited on
Commit
e161171
·
verified ·
1 Parent(s): 40e900b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,7 +62,8 @@ metalatte_path = '.'
62
  sys.path.insert(0, metalatte_path)
63
  from configuration import MetaLATTEConfig
64
  from modeling_metalatte import MultitaskProteinModel
65
-
 
66
  # Load config + instantiate model (no network)
67
  config = AutoConfig.from_pretrained(metalatte_local, local_files_only=True)
68
 
 
62
  sys.path.insert(0, metalatte_path)
63
  from configuration import MetaLATTEConfig
64
  from modeling_metalatte import MultitaskProteinModel
65
+ AutoConfig.register("metalatte", MetaLATTEConfig)
66
+ AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)
67
  # Load config + instantiate model (no network)
68
  config = AutoConfig.from_pretrained(metalatte_local, local_files_only=True)
69