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) |