Update app.py

app.py

@@ -1,16 +1,15 @@
 import streamlit as st
-# from transformers import AutoTokenizer, AutoModelForCausalLM
-from transformers import LlamaTokenizer, LlamaForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load your model here
 # tokenizer = AutoTokenizer.from_pretrained("qbwmwsap/amber-model-mine")
 # model = AutoModelForCausalLM.from_pretrained("qbwmwsap/amber-model-mine")
 
 # Load original amber model
-tokenizer = LlamaTokenizer.from_pretrained("LLM360/Amber", revision="ckpt_356")
-model = LlamaForCausalLM.from_pretrained("LLM360/Amber", revision="ckpt_356")
+tokenizer = AutoTokenizer.from_pretrained("LLM360/Amber")
+model = AutoModelForCausalLM.from_pretrained("LLM360/Amber")
 
-st.title('Amber-Model-Mine')
+st.title('Amber-Model-Remastered')
 
 text = st.text_input('Enter some text')