Solshine commited on
Commit
d603b0b
1 Parent(s): 678dc31

Changed model to squarelike/llama2-ko-medical-7b

Browse files

squarelike/llama2-ko-medical-7b is new model instead of MiniMed

Files changed (1) hide show
  1. MiniMed_EHR_Analyst_Spaces.py +5 -3
MiniMed_EHR_Analyst_Spaces.py CHANGED
@@ -4,10 +4,12 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
4
 
5
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
6
 
7
- # Load the tokenizer and model: pseudolab/K23_MiniMed by Tonic (Note: This is a large model and will take a while to download)
 
 
8
  # Config issues persist with this model, unfortunately. It may not be ready for use.
9
- tokenizer = AutoTokenizer.from_pretrained("pseudolab/K23_MiniMed")
10
- model = AutoModelForCausalLM.from_pretrained("pseudolab/K23_MiniMed")
11
 
12
  #Upload Patient Data
13
  uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
 
4
 
5
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
6
 
7
+ # Load the tokenizer and model:
8
+ # Now is squarelike/llama2-ko-medical-7b
9
+ # formerly pseudolab/K23_MiniMed by Tonic (Note: This is a large model and will take a while to download)
10
  # Config issues persist with this model, unfortunately. It may not be ready for use.
11
+ tokenizer = AutoTokenizer.from_pretrained("squarelike/llama2-ko-medical-7b")
12
+ model = AutoModelForCausalLM.from_pretrained("squarelike/llama2-ko-medical-7b")
13
 
14
  #Upload Patient Data
15
  uploaded_file = st.file_uploader("Choose a CSV file", type="csv")