yuanjunchai commited on
Commit
db9ef6e
·
1 Parent(s): 05fff77

add application files

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,8 +57,8 @@ 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
+ 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)