Neal Caren commited on
Commit
8d60d52
·
1 Parent(s): 4305275
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -23,11 +23,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
 
24
  import pandas as pd
25
 
 
 
26
 
27
  @st.cache
28
- def load_data():
29
- url = st.secrets["graphs_url"]
30
- print(url)
31
  #url = "https://drive.google.com/uc?export=download&id=1nIBS9is8YCeiPBqA7MifVC5xeaKWH8uL"
32
  output = "passages.jsonl"
33
  gdown.download(url, output, quiet=False)
@@ -44,7 +44,7 @@ st.write('This project is a work-in-progress that searches the text of recently-
44
 
45
 
46
  with st.spinner(text="Loading data..."):
47
- df = load_data()
48
  passages = df['text'].values
49
 
50
  no_of_graphs=len(df)
 
23
 
24
  import pandas as pd
25
 
26
+ purl = st.secrets["graphs_url"]
27
+ print(purl)
28
 
29
  @st.cache
30
+ def load_data(url):
 
 
31
  #url = "https://drive.google.com/uc?export=download&id=1nIBS9is8YCeiPBqA7MifVC5xeaKWH8uL"
32
  output = "passages.jsonl"
33
  gdown.download(url, output, quiet=False)
 
44
 
45
 
46
  with st.spinner(text="Loading data..."):
47
+ df = load_data(purl)
48
  passages = df['text'].values
49
 
50
  no_of_graphs=len(df)