ysharma HF staff commited on
Commit
ae8c816
1 Parent(s): c105afe
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def get_answers_timestamp(question, final_transcript, transcript):
60
 
61
  similarity_tensor = util.pytorch_cos_sim(embedding_1, embedding_2)
62
  idx = torch.argmax(similarity_tensor)
63
- start_timestamp = dftranscript.iloc[[int(idx)-1]].start.values[0]
64
  start_timestamp = round(start_timestamp)
65
 
66
  return start_timestamp
 
60
 
61
  similarity_tensor = util.pytorch_cos_sim(embedding_1, embedding_2)
62
  idx = torch.argmax(similarity_tensor)
63
+ start_timestamp = dftranscript.iloc[[int(idx)-3]].start.values[0]
64
  start_timestamp = round(start_timestamp)
65
 
66
  return start_timestamp