Blessmore commited on
Commit
94a4017
·
verified ·
1 Parent(s): 2f254fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -69,7 +69,7 @@ def clean_text_multithreaded(text):
69
 
70
  # Function to load the FastText model from Hugging Face
71
  @st.cache_resource
72
- def load_fasttext_model(model_dir):
73
  #model_path = os.path.join(model_dir, "fasttext_model.model")
74
  #vectors_path = os.path.join(model_dir, "fasttext_model_vectors.kv")
75
  #vectors_ngrams_path = os.path.join(model_dir, "fasttext_model.model.wv.vectors_ngrams.npy")
@@ -207,11 +207,11 @@ def main():
207
  st.header("Generate Embeddings with Pretrained FastText Model")
208
 
209
  repo_id = "Blessmore/Fasttext_embeddings/Fast_text_50_dim"
210
- model_file = "Fast_text_50_dim"
211
  vectors_file = "fasttext_model_vectors.kv"
212
  vectors_ngrams_file = "fasttext_model.model.wv.vectors_ngrams.npy"
213
 
214
- model = load_fasttext_model(model_file)
215
 
216
  st.subheader("Generate Word Embedding")
217
  word = st.text_input("Enter a word:")
 
69
 
70
  # Function to load the FastText model from Hugging Face
71
  @st.cache_resource
72
+ def load_fasttext_model(model_path):
73
  #model_path = os.path.join(model_dir, "fasttext_model.model")
74
  #vectors_path = os.path.join(model_dir, "fasttext_model_vectors.kv")
75
  #vectors_ngrams_path = os.path.join(model_dir, "fasttext_model.model.wv.vectors_ngrams.npy")
 
207
  st.header("Generate Embeddings with Pretrained FastText Model")
208
 
209
  repo_id = "Blessmore/Fasttext_embeddings/Fast_text_50_dim"
210
+ model_path = "Fast_text_50_dim"
211
  vectors_file = "fasttext_model_vectors.kv"
212
  vectors_ngrams_file = "fasttext_model.model.wv.vectors_ngrams.npy"
213
 
214
+ model = load_fasttext_model(model_path)
215
 
216
  st.subheader("Generate Word Embedding")
217
  word = st.text_input("Enter a word:")