Sirinoot commited on
Commit
c423f23
·
verified ·
1 Parent(s): 115ed91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -157,9 +157,11 @@ def predict_test(model, tokenizer, embedding_model, df, question, index): # sen
157
 
158
 
159
  segments = sent_tokenize(mostSimContext, engine="crfcut")
 
160
  segments_index = set_index(get_embeddings(embedding_model,segments))
161
  _distances,_indices = faiss_search(segments_index, question_vector)
162
  mostSimSegment = segments[_indices[0][0]]
 
163
 
164
  Answer = model_pipeline(model, tokenizer,question,mostSimSegment)
165
 
 
157
 
158
 
159
  segments = sent_tokenize(mostSimContext, engine="crfcut")
160
+ print('segments',segments)
161
  segments_index = set_index(get_embeddings(embedding_model,segments))
162
  _distances,_indices = faiss_search(segments_index, question_vector)
163
  mostSimSegment = segments[_indices[0][0]]
164
+ print('mostSimSegment',mostSimSegment)
165
 
166
  Answer = model_pipeline(model, tokenizer,question,mostSimSegment)
167