Sirinoot commited on
Commit
28ae8db
·
verified ·
1 Parent(s): 31bc28f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ def predict_test(model, tokenizer, embedding_model, df, question, index): # sen
162
 
163
  segments = sent_tokenize(mostSimContext, engine="crfcut")
164
 
165
- segment_embeddings = get_embeddings(segments)
166
  segment_embeddings = prepare_sentences_vector(segment_embeddings)
167
  segment_index = create_segment_index(segment_embeddings)
168
 
 
162
 
163
  segments = sent_tokenize(mostSimContext, engine="crfcut")
164
 
165
+ segment_embeddings = get_embeddings(embedding_model, segments)
166
  segment_embeddings = prepare_sentences_vector(segment_embeddings)
167
  segment_index = create_segment_index(segment_embeddings)
168