Update app.py
Browse files
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-
|
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()
|