Darsh1234Tayal commited on
Commit
fe659d4
·
verified ·
1 Parent(s): 10042d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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, persist_directory='chroma_db')
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(inbrowser=True)
 
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()