will.k
commited on
Commit
•
3a4a22b
1
Parent(s):
80a2449
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
|