Update app.py

app.py

@@ -6,14 +6,14 @@ from peft import PeftModel, PeftConfig
 #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
 
 # Instantiate the Tokenizer
-tokenizer = AutoTokenizer.from_pretrained("squarelike/llama2-ko-medical-7b", trust_remote_code=True, padding_side="left")
+tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True, padding_side="left")
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = 'left'
 
 # Load the PEFT model
-peft_config = PeftConfig.from_pretrained("squarelike/llama2-ko-medical-7b")
-peft_model = MistralForCausalLM.from_pretrained("squarelike/llama2-ko-medical-7b", trust_remote_code=True)
-peft_model = PeftModel.from_pretrained(peft_model, "squarelike/llama2-ko-medical-7b")
+peft_config = PeftConfig.from_pretrained("pseudolab/K23_MiniMed")
+peft_model = MistralForCausalLM.from_pretrained("pseudolab/K23_MiniMed", trust_remote_code=True)
+peft_model = PeftModel.from_pretrained(peft_model, "pseudolab/K23_MiniMed")
 
 #Upload Patient Data
 uploaded_file = st.file_uploader("Choose a CSV file", type="csv")