yuanjunchai commited on
Commit ·
6990f7c
1
Parent(s): db9ef6e
add application files
Browse files
app.py
CHANGED
|
@@ -57,8 +57,9 @@ def download_glove_embeddings_gdrive(model_type):
|
|
| 57 |
word_index_temp = "word_index_dict_" + str(model_type) + "_temp.pkl"
|
| 58 |
# 100d download
|
| 59 |
if model_type == "100d":
|
| 60 |
-
hf_hub_download(repo_id='AveMujica/glove-twitter-100d', filename='embeddings_100d_temp.npy'
|
| 61 |
-
hf_hub_download(repo_id='AveMujica/glove-twitter-100d', filename='word_index_dict_100d_temp.pkl'
|
|
|
|
| 62 |
else:
|
| 63 |
# Get glove embeddings from google drive
|
| 64 |
word_index_id, embeddings_id = get_model_id_gdrive(model_type)
|
|
|
|
| 57 |
word_index_temp = "word_index_dict_" + str(model_type) + "_temp.pkl"
|
| 58 |
# 100d download
|
| 59 |
if model_type == "100d":
|
| 60 |
+
file1_path = hf_hub_download(repo_id='AveMujica/glove-twitter-100d', filename='embeddings_100d_temp.npy')
|
| 61 |
+
file2_path = hf_hub_download(repo_id='AveMujica/glove-twitter-100d', filename='word_index_dict_100d_temp.pkl')
|
| 62 |
+
print(f"File downloaded to: {file1_path}, {file2_path}")
|
| 63 |
else:
|
| 64 |
# Get glove embeddings from google drive
|
| 65 |
word_index_id, embeddings_id = get_model_id_gdrive(model_type)
|