Sirinoot commited on
Commit
8f3c617
·
verified ·
1 Parent(s): df4f1f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -171,8 +171,8 @@ def predict_test(model, tokenizer, embedding_model, df, question, index): # sen
171
 
172
  Answer = model_pipeline(model, tokenizer,question,mostSimSegment)
173
 
174
- if len(answer) <= 2:
175
- answer = mostSimSegment
176
 
177
  # Find the start and end indices of mostSimSegment within mostSimContext
178
  start_index = mostSimContext.find(Answer)
 
171
 
172
  Answer = model_pipeline(model, tokenizer,question,mostSimSegment)
173
 
174
+ if len(Answer) <= 2:
175
+ Answer = mostSimSegment
176
 
177
  # Find the start and end indices of mostSimSegment within mostSimContext
178
  start_index = mostSimContext.find(Answer)