Update app.py

app.py

@@ -1,5 +1,10 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModel, AutoConfig
+metalatte_path = './Chatterjeelab/MetaLATTE'
+sys.path.insert(0, metalatte_path)
+from metalatte import MetaLATTEConfig, MetaLATTEModel
+AutoConfig.register("metalatte", MetaLATTEConfig)
+AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
 
 tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
 config = AutoConfig.from_pretrained("ChatterjeeLab/MetaLATTE")