Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import gradio as gr
|
|
| 7 |
from dotenv import load_dotenv
|
| 8 |
import os
|
| 9 |
from urllib.parse import urlparse, parse_qs
|
|
|
|
| 10 |
|
| 11 |
api_key = os.environ.get("BYTEZ_API_KEY")
|
| 12 |
sdk = Bytez(api_key)
|
|
@@ -46,7 +47,7 @@ def create_and_save_vs(trans):
|
|
| 46 |
splitter = RecursiveCharacterTextSplitter(chunk_size = 100, chunk_overlap = 50)
|
| 47 |
docs = splitter.split_text(trans)
|
| 48 |
embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2')
|
| 49 |
-
vector_store_db = Chroma.from_texts(docs, embeddings
|
| 50 |
except Exception:
|
| 51 |
return None
|
| 52 |
return vector_store_db
|
|
@@ -384,4 +385,4 @@ with gr.Blocks(
|
|
| 384 |
submit_button.click(setter, inputs=[youtube_link], outputs=[first_page, loading_page, chat_page, wrong_link_page, cc_not_enabled, normal_error, summary, vs])
|
| 385 |
submit_answer.click(execute, inputs=[vs, ques], outputs=[answer, chat_page, normal_error])
|
| 386 |
|
| 387 |
-
ui.launch(
|
|
|
|
| 7 |
from dotenv import load_dotenv
|
| 8 |
import os
|
| 9 |
from urllib.parse import urlparse, parse_qs
|
| 10 |
+
import time
|
| 11 |
|
| 12 |
api_key = os.environ.get("BYTEZ_API_KEY")
|
| 13 |
sdk = Bytez(api_key)
|
|
|
|
| 47 |
splitter = RecursiveCharacterTextSplitter(chunk_size = 100, chunk_overlap = 50)
|
| 48 |
docs = splitter.split_text(trans)
|
| 49 |
embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2')
|
| 50 |
+
vector_store_db = Chroma.from_texts(docs, embeddings)
|
| 51 |
except Exception:
|
| 52 |
return None
|
| 53 |
return vector_store_db
|
|
|
|
| 385 |
submit_button.click(setter, inputs=[youtube_link], outputs=[first_page, loading_page, chat_page, wrong_link_page, cc_not_enabled, normal_error, summary, vs])
|
| 386 |
submit_answer.click(execute, inputs=[vs, ques], outputs=[answer, chat_page, normal_error])
|
| 387 |
|
| 388 |
+
ui.launch()
|