Update app.py

app.py

@@ -5,8 +5,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 os.environ["HF_HOME"] = "/data/hf_cache"
 
-tokenizer = AutoTokenizer.from_pretrained("/data/model")
-model = AutoModelForCausalLM.from_pretrained("/data/model", device_map="auto", offload_folder="/tmp/offload")
+tokenizer = AutoTokenizer.from_pretrained("/data/model1")
+model = AutoModelForCausalLM.from_pretrained("/data/model1", device_map="auto", offload_folder="/tmp/offload")
 
 def get_stats():
     process = psutil.Process(os.getpid())