Spaces:
Build error
Build error
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)-
|
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
|