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