Solshine commited on
Commit
512eba0
1 Parent(s): b99b31d

Update app.py

Browse files

Switching theme function to Streamlit because gradio still gave error stating gr not defined

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import streamlit as st
2
  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
 
7
- gr.set_theme(custom_theme="pseudolab/huggingface-korea-theme'")
8
 
9
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
10
 
 
1
  import streamlit as st
2
  import pandas as pd
3
  from transformers import pipeline, AutoConfig, AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, MistralForCausalLM
4
+ from peft import PeftModel, PeftConfigimport streamlit as st
5
+ from streamlit_theme import theme
6
 
7
+ st.set_theme("pseudolab/huggingface-korea-theme')
8
 
9
  #Note this should be used always in compliance with applicable laws and regulations if used with real patient data.
10