Update app.py

app.py

@@ -17,11 +17,11 @@ st.title("Experimental KPhi3 Model - Currently in Training")
 
 # Load tokenizer and model
 @st.cache_resource(show_spinner="Loading model...")
-def load_model(repo_name):
-    # tokenizer = AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
-    tokenizer = LlamaTokenizer.from_pretrained(repo_name, trust_remote_code=True)
-    generator_conf = GenerationConfig.from_pretrained(repo_name)
-    model = AutoModelForCausalLM.from_pretrained(repo_name, trust_remote_code=True, torch_dtype=torch.bfloat16)
+def load_model(local_repo_name):
+    # tokenizer = AutoTokenizer.from_pretrained(local_repo_name, trust_remote_code=True)
+    tokenizer = LlamaTokenizer.from_pretrained(local_repo_name, trust_remote_code=True)
+    generator_conf = GenerationConfig.from_pretrained(local_repo_name)
+    model = AutoModelForCausalLM.from_pretrained(local_repo_name, trust_remote_code=True, torch_dtype=torch.bfloat16)
     return tokenizer, generator_conf, model
 
 tokenizer, generator_conf, model = load_model(REPO_NAME)