Update app.py

app.py

@@ -1,8 +1,16 @@
 import spaces
-from transformers import pipeline
 import gradio as gr
-model = 'mosaicml/mpt-7b-instruct'
-tokenizer = 'mosaicml/mpt-7b-instruct'
+import transformers
+from transformers import AutoTokenizer
+from transformers import pipeline
+
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b")
+
+model = transformers.AutoModelForCausalLM.from_pretrained(
+  'mosaicml/mpt-7b-instruct',
+  trust_remote_code=True
+)
+
 
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')