Solshine commited on
Commit
cac308b
1 Parent(s): ba57ab6

Update MiniMed_EHR_Analyst_Spaces.py

Browse files
Files changed (1) hide show
  1. MiniMed_EHR_Analyst_Spaces.py +3 -4
MiniMed_EHR_Analyst_Spaces.py CHANGED
@@ -5,11 +5,10 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
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")
 
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 pseudolab/K23_MiniMed by Tonic (Note: This model has been struggling to operate without config errors)
 
9
  # Config issues persist with this model, unfortunately. It may not be ready for use.
10
+ tokenizer = AutoTokenizer.from_pretrained("AutoTokenizer.from_pretrained("pseudolab/K23_MiniMed)
11
+ model = AutoModelForCausalLM.from_pretrained("pseudolab/K23_MiniMed)
12
 
13
  #Upload Patient Data
14
  uploaded_file = st.file_uploader("Choose a CSV file", type="csv")