yuanjunchai commited on
Commit
6990f7c
·
1 Parent(s): db9ef6e

add application files

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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', local_dir=embeddings_temp)
61
- hf_hub_download(repo_id='AveMujica/glove-twitter-100d', filename='word_index_dict_100d_temp.pkl', local_dir=word_index_temp)
 
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)