georad commited on
Commit
c684f94
·
verified ·
1 Parent(s): 137dd16

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import streamlit as st
3
+ #import torch
4
+
5
+
6
+ def get_device_map() -> str:
7
+ return 'cuda' if torch.cuda.is_available() else 'cpu'
8
+ device = get_device_map() # 'cpu'
9
+
10
+ def setup_page()
11
+ with st.spinner("In progress...):
12
+ hf_token = os.getenv('HF_TOKEN')
13
+ df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv")
14
+
15
+ if __name__ == "__main__":
16
+ st.set_page_config(page_title="Sticky Header test", layout="wide")
17
+ setup_page()