MoritzMMuller commited on
Commit
9e906f9
·
verified ·
1 Parent(s): a2ce149

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -4
src/streamlit_app.py CHANGED
@@ -1,4 +1,3 @@
1
- %%writefile saved_models/app.py
2
  import os
3
  import streamlit as st
4
  from PIL import Image
@@ -30,8 +29,7 @@ LLM_NAME = "google/flan-t5-xl"
30
  HF_TOKEN = ".."
31
  DATA_DIR = "data/harvard_dataset" # Path where you download and unpack the Harvard Dataverse dataset
32
  DIARY_CSV = "diary.csv"
33
- CANCER_DIR = r"D:\Models\googleflan-t5-xl"
34
- LLM_DIR = r"D:\Models\SkinCancer"
35
 
36
  # Initialize session state defaults
37
  if 'initialized' not in st.session_state:
@@ -335,4 +333,4 @@ st.sidebar.write(f"Model: {MODEL_NAME}")
335
  st.sidebar.write(f"LLM: {LLM_NAME}")
336
 
337
  if __name__ == '__main__':
338
- st.write()
 
 
1
  import os
2
  import streamlit as st
3
  from PIL import Image
 
29
  HF_TOKEN = ".."
30
  DATA_DIR = "data/harvard_dataset" # Path where you download and unpack the Harvard Dataverse dataset
31
  DIARY_CSV = "diary.csv"
32
+
 
33
 
34
  # Initialize session state defaults
35
  if 'initialized' not in st.session_state:
 
333
  st.sidebar.write(f"LLM: {LLM_NAME}")
334
 
335
  if __name__ == '__main__':
336
+ st.write()