will.k commited on
Commit
3a4a22b
1 Parent(s): 80a2449
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -3,7 +3,6 @@ import pandas as pd
3
  from transformers import pipeline, AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
4
  from peft import PeftModel, PeftConfig
5
  import gradio as gr
6
- gr.Blocks(theme= 'pseudolab/huggingface-korea-theme')
7
 
8
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
9
 
@@ -17,9 +16,6 @@ peft_config = PeftConfig.from_pretrained("pseudolab/K23_MiniMed")
17
  peft_model = MistralForCausalLM.from_pretrained("pseudolab/K23_MiniMed", trust_remote_code=True)
18
  peft_model = PeftModel.from_pretrained(peft_model, "pseudolab/K23_MiniMed")
19
 
20
- #Upload Patient Data
21
- uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
22
-
23
  # Prepare the context
24
  def prepare_context(data):
25
  # Format the data as a string
 
3
  from transformers import pipeline, AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
4
  from peft import PeftModel, PeftConfig
5
  import gradio as gr
 
6
 
7
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
8
 
 
16
  peft_model = MistralForCausalLM.from_pretrained("pseudolab/K23_MiniMed", trust_remote_code=True)
17
  peft_model = PeftModel.from_pretrained(peft_model, "pseudolab/K23_MiniMed")
18
 
 
 
 
19
  # Prepare the context
20
  def prepare_context(data):
21
  # Format the data as a string