Spaces:
Runtime error
Runtime error
File size: 815 Bytes
69f3d2a 5489720 69f3d2a 32c50ef 5489720 5d7f105 487553a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import pinecone
import streamlit as st
from sentence_transformers import SentenceTransformer
index_id = "youtube-search"
pinecone.init(
api_key='e815acaa-c5d2-4280-85bc-4994c59a5ee4', # app.pinecone.io
environment="us-west1-gcp" # find next to API key
)
index = pinecone.Index(index_id)
query = st.text_input("Query (exmaple: when to raise funds?)")
if query:
model_id = "multi-qa-mpnet-base-dot-v1"
model = SentenceTransformer(model_id)
xq = model.encode(query).tolist()
output = index.query(xq, top_k=5, include_metadata=True)
for o in output["matches"]:
url = o["id"].split("-")[0]
start = o["metadata"]["start"]
url_time = "https://youtu.be/" + url+ f"?t={start}"
text = o["metadata"]["text"]
st.write(url_time)
st.write(text) |