Update app.py

app.py

@@ -2,12 +2,8 @@ import streamlit as st
 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 = AutoTokenizer.from_pretrained("LLM360/Amber")
-model = AutoModelForCausalLM.from_pretrained("LLM360/Amber")
+tokenizer = AutoTokenizer.from_pretrained("qbwmwsap/amber-model-mine")
+model = AutoModelForCausalLM.from_pretrained("qbwmwsap/amber-model-mine")
 
 st.title('Amber-Model-Remastered')