EdwardXu commited on
Commit
a07cb6e
1 Parent(s): de6e3d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -19,9 +19,9 @@ drive.mount('/content/drive')
19
 
20
  """## Install dependencies"""
21
 
22
- !pip3 install sentence-transformers
23
- !pip install datasets
24
- !pip install -q streamlit
25
 
26
  """## 1. Embeddings
27
 
@@ -52,13 +52,13 @@ class Embeddings:
52
  """
53
  Download glove embeddings from web or from your gdrive if in optimized format
54
  """
55
- embeddings_temp = "/content/drive/MyDrive/EE596LLM/HW2/embeddings_50d_temp.npy"
56
- word_index_temp = "/content/drive/MyDrive/EE596LLM/HW2/word_index_dict_50d_temp.pkl"
57
 
58
 
59
  def load_glove_embeddings(self, embedding_dimension):
60
- word_index_temp = "/content/drive/MyDrive/EE596LLM/HW2/word_index_dict_50d_temp.pkl"
61
- embeddings_temp = "/content/drive/MyDrive/EE596LLM/HW2/embeddings_50d_temp.npy"
62
 
63
  # Load word index dictionary
64
  word_index_dict = pickle.load(open(word_index_temp, "rb"), encoding="latin")
 
19
 
20
  """## Install dependencies"""
21
 
22
+ #!pip3 install sentence-transformers
23
+ #!pip install datasets
24
+ #!pip install -q streamlit
25
 
26
  """## 1. Embeddings
27
 
 
52
  """
53
  Download glove embeddings from web or from your gdrive if in optimized format
54
  """
55
+ embeddings_temp = "embeddings_50d_temp.npy"
56
+ word_index_temp = "word_index_dict_50d_temp.pkl"
57
 
58
 
59
  def load_glove_embeddings(self, embedding_dimension):
60
+ word_index_temp = "word_index_dict_50d_temp.pkl"
61
+ embeddings_temp = "embeddings_50d_temp.npy"
62
 
63
  # Load word index dictionary
64
  word_index_dict = pickle.load(open(word_index_temp, "rb"), encoding="latin")