Update app.py

app.py

@@ -7,9 +7,9 @@ sys.path.insert(0, metalatte_path)
 
 # Import the custom configuration and model
 from configuration import MetaLATTEConfig
-from modeling_metalatte  import MetaLATTEModel
+from modeling_metalatte  import MultitaskProteinModel
 AutoConfig.register("metalatte", MetaLATTEConfig)
-AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
+AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)
 
 tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
 config = AutoConfig.from_pretrained("ChatterjeeLab/MetaLATTE")