Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 import sys
 from transformers import AutoTokenizer, AutoModel, AutoConfig
-metalatte_path = './Chatterjeelab/MetaLATTE'
+# Add the current directory to the system path
+metalatte_path = '.'
 sys.path.insert(0, metalatte_path)
-from metalatte import MetaLATTEConfig, MetaLATTEModel
+
+# Import the custom configuration and model
+from configuration import MetaLATTEConfig
+from model import MetaLATTEModel
 AutoConfig.register("metalatte", MetaLATTEConfig)
 AutoModel.register(MetaLATTEConfig, MetaLATTEModel)