dsleo commited on
Commit
c28a67e
·
verified ·
1 Parent(s): 3e140a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import pandas as pd
3
  import json
4
  import os
5
- import gzip
6
  from sentence_transformers import SentenceTransformer, util
7
  from loguru import logger
8
 
@@ -17,8 +17,10 @@ model = SentenceTransformer(MODEL_NAME)
17
  @st.cache_data
18
  def load_data():
19
  file_path = "data/merged_dataset.csv.zip"
20
- with gzip.open(file_path, "rt") as f:
21
- df = pd.read_csv(f)
 
 
22
  return df
23
 
24
  df = load_data()
 
2
  import pandas as pd
3
  import json
4
  import os
5
+ import zipfile
6
  from sentence_transformers import SentenceTransformer, util
7
  from loguru import logger
8
 
 
17
  @st.cache_data
18
  def load_data():
19
  file_path = "data/merged_dataset.csv.zip"
20
+ with zipfile.ZipFile(file_path, 'r') as zip_ref:
21
+ zip_ref.printdir()
22
+ zip_ref.extractall("data/extracted")
23
+ df = pd.read_csv("data/extracted/merged_dataset.csv")
24
  return df
25
 
26
  df = load_data()