Kathir0011 commited on
Commit
741e534
·
verified ·
1 Parent(s): 775bf34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def create_db_from_video_url(video_url, api_key):
16
  """
17
  Creates an Embedding of the Video and performs
18
  """
19
- embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=api_key)
20
 
21
  loader = YoutubeLoader.from_youtube_url(video_url)
22
  transcripts = loader.load()
 
16
  """
17
  Creates an Embedding of the Video and performs
18
  """
19
+ embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004", google_api_key=api_key)
20
 
21
  loader = YoutubeLoader.from_youtube_url(video_url)
22
  transcripts = loader.load()