Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 @st.cache(allow_output_mutation=True)
 def load_model():
@@ -7,9 +7,9 @@ def load_model():
     Loads the model and tokenizer from the local directory.
     :return: A list containing the model and the tokenizer.
     '''
-    model_name = 'WIP'
-    tokenizer = GPT2Tokenizer.from_pretrained(model_name)
-    model = GPT2LMHeadModel.from_pretrained(model_name)
+    model_name = 'facebook/incoder-6B'
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
     return [model, tokenizer]
 
 st.set_page_config(